self.root is Path or None + The following packages are already present in the pyproject.toml and will be skipped:

- huggingface-hub

If you want to update it to the latest compatible version, you can use `poetry update package`.
If you prefer to upgrade it to the latest available version, you can use `poetry add package@latest`.

Nothing to add.
This commit is contained in:
Cadene
2024-03-15 10:56:46 +00:00
parent b10c9507d4
commit 41521f7e96
6 changed files with 15 additions and 13 deletions

View File

@@ -112,7 +112,8 @@ class PushtExperienceReplay(AbstractExperienceReplay):
)
def _download_and_preproc_obsolete(self):
raw_dir = Path(self.root) / f"{self.dataset_id}_raw"
assert self.root is not None
raw_dir = self.root / f"{self.dataset_id}_raw"
zarr_path = (raw_dir / PUSHT_ZARR).resolve()
if not zarr_path.is_dir():
raw_dir.mkdir(parents=True, exist_ok=True)
@@ -208,7 +209,7 @@ class PushtExperienceReplay(AbstractExperienceReplay):
if episode_id == 0:
# hack to initialize tensordict data structure to store episodes
td_data = ep_td[0].expand(total_frames).memmap_like(Path(self.root) / f"{self.dataset_id}")
td_data = ep_td[0].expand(total_frames).memmap_like(self.root / f"{self.dataset_id}")
td_data[idxtd : idxtd + len(ep_td)] = ep_td