Clean logging, Refactor

lerobot/common/datasets/pusht.py

@@ -177,6 +177,14 @@ class PushtExperienceReplay(TensorDictReplayBuffer):
             transform=transform,
         )
 
+    @property
+    def num_samples(self):
+        return len(self)
+
+    @property
+    def num_episodes(self):
+        return len(self._storage._storage["episode"].unique())
+
     @property
     def data_path_root(self):
         if self.streaming:

lerobot/common/datasets/simxarm.py

@@ -109,6 +109,14 @@ class SimxarmExperienceReplay(TensorDictReplayBuffer):
             transform=transform,
         )
 
+    @property
+    def num_samples(self):
+        return len(self)
+
+    @property
+    def num_episodes(self):
+        return len(self._storage._storage["episode"].unique())
+
     @property
     def data_path_root(self):
         if self.streaming: