Remove noqa-F821

2024-03-05 10:22:21 +00:00
parent d4e0849970
commit 2f80d71c3e
2 changed files with 2 additions and 2 deletions
--- a/lerobot/common/datasets/abstract.py
+++ b/lerobot/common/datasets/abstract.py
@@ -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
--- a/lerobot/common/datasets/pusht.py
+++ b/lerobot/common/datasets/pusht.py
@@ -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,