Update

[ghstack-poisoned]

tensordict/base.py

@@ -18,6 +18,7 @@
 import uuid
 import warnings
 import weakref
+from collections import UserDict
 from collections.abc import MutableMapping
 
 from concurrent.futures import Future, ThreadPoolExecutor, wait
@@ -9871,6 +9872,8 @@ def from_any(cls, obj, *, auto_batch_size: bool = False):
             return obj
         if isinstance(obj, dict):
             return cls.from_dict(obj, auto_batch_size=auto_batch_size)
+        if isinstance(obj, UserDict):
+            return cls.from_dict(dict(obj), auto_batch_size=auto_batch_size)
         if isinstance(obj, np.ndarray) and hasattr(obj.dtype, "names"):
             return cls.from_struct_array(obj, auto_batch_size=auto_batch_size)
         if isinstance(obj, tuple):

test/test_tensordict.py

@@ -18,6 +18,7 @@
 import sys
 import uuid
 import warnings
+from collections import UserDict
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Any
@@ -996,6 +997,13 @@ class MyClass:
             td.keys(True, True)
         ).symmetric_difference(expected)
 
+    def test_from_any_userdict(self):
+        class D(UserDict): ...
+
+        d = D(a=0)
+        assert TensorDict.from_any(d)["a"] == 0
+        assert isinstance(TensorDict.from_any(d)["a"], torch.Tensor)
+
     def test_from_dataclass(self):
         @dataclass
         class MyClass: