From 2f80d71c3e8b4ccf48d232e354baa1d8af6236f4 Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Tue, 5 Mar 2024 10:22:21 +0000 Subject: [PATCH] Remove noqa-F821 --- lerobot/common/datasets/abstract.py | 2 +- lerobot/common/datasets/pusht.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/common/datasets/abstract.py b/lerobot/common/datasets/abstract.py index 0407c4f6..4f56d96f 100644 --- 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 diff --git a/lerobot/common/datasets/pusht.py b/lerobot/common/datasets/pusht.py index e086df27..77334851 100644 --- 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,