tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

use Path type instead of str

Browse Source
This commit is contained in:
Alexander Soare
2024-03-15 13:15:34 +00:00
parent 9c88071bc7
commit d374873849
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/abstract.py
View File

@@ -85,7 +85,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer):
self._transform = transform self._transform = transform
def compute_or_load_stats(self, num_batch=100, batch_size=32) -> TensorDict: 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(): if stats_path.exists():
stats = torch.load(stats_path) stats = torch.load(stats_path)
else: else: