wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove noqa-F821

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,