Update

Vincent Moens · Vincent Moens · commit 5149a5173e12 · 2025-03-18T07:27:55.000-07:00
[ghstack-poisoned]
diff --git a/tensordict/nn/probabilistic.py b/tensordict/nn/probabilistic.py
@@ -1361,7 +1361,7 @@ def forward(
                     ]
                 else:
                     keys = list(set(self.out_keys + list(tensordict.keys(True, True))))
-                return tensordict_out.update(result, keys_to_update=keys)
+                return tensordict.update(result, keys_to_update=keys)
         return result
 
 

Original file line number	Diff line number	Diff line change
`@@ -1361,7 +1361,7 @@ def forward(`
`1361`	`1361`	`]`
`1362`	`1362`	`else:`
`1363`	`1363`	`keys = list(set(self.out_keys + list(tensordict.keys(True, True))))`
`1364`		`- return tensordict_out.update(result, keys_to_update=keys)`
	`1364`	`+ return tensordict.update(result, keys_to_update=keys)`
`1365`	`1365`	`return result`
`1366`	`1366`
`1367`	`1367`