lint

pytorch · Nov 21, 2023 · 86e33ac · 86e33ac
1 parent a4d70cb
commit 86e33ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensordict/base.py b/tensordict/base.py
@@ -528,7 +528,7 @@ def chunk(self, chunks: int, dim: int = 0) -> tuple[TensorDictBase, ...]:
                 f"chunks must be a strictly positive integer, got {chunks}."
             )
         # fall back on split, using upper rounding
-        split_size = - (self.batch_size[dim] // -chunks)
+        split_size = -(self.batch_size[dim] // -chunks)
         return self.split(split_size, dim=dim)
 
     @overload