use Path type instead of str

2024-03-15 13:15:34 +00:00
parent 9c88071bc7
commit d374873849
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/abstract.py
+++ b/lerobot/common/datasets/abstract.py
@@ -85,7 +85,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer):
            self._transform = transform

    def compute_or_load_stats(self, num_batch=100, batch_size=32) -> TensorDict:
-        stats_path = self.data_dir / "stats.pth"
+        stats_path = Path(self.data_dir) / "stats.pth"
        if stats_path.exists():
            stats = torch.load(stats_path)
        else: