Episode lenght now returns array

boris-il-forte · boris-il-forte · commit 77affeeaa313 · 2024-01-22T16:23:14.000+01:00
- now episode length returns backend array
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):
