From 951e4e985656f3e9885f01415c8596bbcb679121 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Tue, 10 Sep 2024 16:27:26 +0100
Subject: [PATCH] Update

[ghstack-poisoned]
---
 tensordict/tensorclass.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensordict/tensorclass.py b/tensordict/tensorclass.py
index 8d6abde1d..f28094e48 100644
--- 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)