amend

pytorch · Jun 7, 2024 · 39a44fb · 39a44fb
1 parent 13ec40b
commit 39a44fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/data/replay_buffers/samplers.py b/torchrl/data/replay_buffers/samplers.py
@@ -1747,7 +1747,7 @@ def _padded_indices(self, shapes, arange) -> torch.Tensor:
         return pad
 
     @implement_for("torch", None, "2.4")
-    def _padded_indices(self, shapes, arange) -> torch.Tensor:  # F811
+    def _padded_indices(self, shapes, arange) -> torch.Tensor:  # noqa: F811
         arange = arange.flip(0).split(shapes.flip(0).squeeze().unbind())
         return (
             torch.nn.utils.rnn.pad_sequence(arange, batch_first=True, padding_value=-1)