Episode lenght now returns array

- now episode length returns backend array
MushroomRL · Jan 22, 2024 · 77affee · 77affee
1 parent 5dc8422
commit 77affee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mushroom_rl/core/dataset.py b/mushroom_rl/core/dataset.py
@@ -291,7 +291,7 @@ def episodes_length(self):
                 lengths.append(l)
                 l = 0
 
-        return lengths
+        return self._array_backend.from_list(lengths)
 
     @property
     def n_episodes(self):