Skip to content

Commit fc55866

Browse files
committed
[CI] Fix ufmt formatting in tensorclass.py
Made-with: Cursor
1 parent a721594 commit fc55866

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensordict/tensorclass.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1551,7 +1551,9 @@ def is_tensor_or_optional_tensor(type_hint):
15511551
)
15521552
# Handle parameterized generics (e.g., TensorDict[str, Tensor])
15531553
if origin is not None and isinstance(origin, type):
1554-
return issubclass(origin, _TensorTypes) or _is_tensor_collection(origin)
1554+
return issubclass(origin, _TensorTypes) or _is_tensor_collection(
1555+
origin
1556+
)
15551557
return False
15561558

15571559
for key, val in cls._type_hints.items():

0 commit comments

Comments
 (0)