Update

[ghstack-poisoned]
pytorch · Sep 10, 2024 · 951e4e9 · 951e4e9
1 parent 525d0de
commit 951e4e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensordict/tensorclass.py b/tensordict/tensorclass.py
@@ -76,12 +76,12 @@
     from typing import dataclass_transform
 except ImportError:
 
-    def dataclass_transform(args):
+    def dataclass_transform(*args, **kwargs):
         """No-op.
 
         Placeholder for dataclass_transform (python<3.11).
         """
-        return args
+        return lambda cls: cls
 
 
 T = TypeVar("T", bound=TensorDictBase)