From d374873849613536599a6174a9f622cb17474779 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Fri, 15 Mar 2024 13:15:34 +0000 Subject: [PATCH] use Path type instead of str --- lerobot/common/datasets/abstract.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/abstract.py b/lerobot/common/datasets/abstract.py index e96133103..bbd7bd285 100644 --- 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: