[BugFix] update TensorDictModule out_keys on assignment #2922

Job	Run time
generate-matrix / generate	7s
pytorch/tensordict (pytorch/tensordict, test/smoke_test.py, .github/scripts/linux-pre-script.sh, ... / build-manywheel-py3_10-cpu	6m 34s
pytorch/tensordict (pytorch/tensordict, test/smoke_test.py, .github/scripts/linux-pre-script.sh, ... / upload / upload-manywheel-py3_10-cpu	28s
	7m 9s

Navigation Menu