[BugFix] Fix none ref in during reduction (#1090)

(cherry picked from commit c11024e)
pytorch · Nov 14, 2024 · 88c86f8 · 88c86f8
1 parent cdd5cb3
commit 88c86f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensordict/base.py b/tensordict/base.py
@@ -10687,7 +10687,7 @@ def set_(x):
                     tgt = mb_unwrap_functional_tensor(new_thing)
                     src = mb_unwrap_functional_tensor(ragged_source)
                     tgt.nested_int_memo = src.nested_int_memo
-                else:
+                elif new_thing is not None:
                     _tensor_symint_registry[new_thing] = _tensor_symint_registry[
                         ragged_source
                     ]