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

vmoens · vmoens · commit 88c86f8379bc · 2024-11-14T13:59:31.000Z
(cherry picked from commit c11024e)
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
                     ]

Original file line number	Diff line number	Diff line change
`@@ -10687,7 +10687,7 @@ def set_(x):`
`10687`	`10687`	`tgt = mb_unwrap_functional_tensor(new_thing)`
`10688`	`10688`	`src = mb_unwrap_functional_tensor(ragged_source)`
`10689`	`10689`	`tgt.nested_int_memo = src.nested_int_memo`
`10690`		`- else:`
	`10690`	`+ elif new_thing is not None:`
`10691`	`10691`	`_tensor_symint_registry[new_thing] = _tensor_symint_registry[`
`10692`	`10692`	`ragged_source`
`10693`	`10693`	`]`