chao.wu/Lerobot
1
0
Fork 0
Code Pull Requests Activity

Remove noqa-F821

Browse Source
This commit is contained in:
Remi Cadene 2024-03-05 10:22:21 +00:00
parent d4e0849970
commit 2f80d71c3e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -29,7 +29,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer):
sampler: SliceSampler = None,
collate_fn: Callable = None,
writer: Writer = None,
transform: "torchrl.envs.Transform" = None, # noqa-F821
transform: "torchrl.envs.Transform" = None,
):
self.dataset_id = dataset_id
self.shuffle = shuffle

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

@ -95,7 +95,7 @@ class PushtExperienceReplay(AbstractExperienceReplay):
sampler: SliceSampler = None,
collate_fn: Callable = None,
writer: Writer = None,
transform: "torchrl.envs.Transform" = None, # noqa-F821
transform: "torchrl.envs.Transform" = None,
):
super().__init__(
dataset_id,