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,