Update

[ghstack-poisoned]
pytorch · Feb 6, 2025 · 6f84874 · 6f84874
1 parent 0220471
commit 6f84874
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tensordict/base.py b/tensordict/base.py
@@ -6444,7 +6444,6 @@ def _get_tuple_maybe_non_tensor(self, key, default):
             return result.data
         return result
 
-
     @overload
     def get_at(self, key, index): ...
 

diff --git a/tensordict/tensorclass.pyi b/tensordict/tensorclass.pyi
@@ -601,11 +601,14 @@ class TensorClass:
     def get(self, key, default): ...
     def get(self, key: NestedKey, *args, **kwargs) -> CompatibleType: ...
     @overload
-    def get_at(self, key, index):...
+    def get_at(self, key, index): ...
     @overload
     def get_at(self, key, index, default): ...
     def get_at(
-        self, key: NestedKey, *args, **kwargs,
+        self,
+        key: NestedKey,
+        *args,
+        **kwargs,
     ) -> CompatibleType: ...
     def get_item_shape(self, key: NestedKey): ...
     def update(