We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
none
1 parent 539651f commit 2f60f8dCopy full SHA for 2f60f8d
tensordict/utils.py
@@ -2214,6 +2214,8 @@ def capture_non_tensor_stack(allow_none=False):
2214
return None
2215
elif _CAPTURE_NONTENSOR_STACK is None:
2216
return _DEFAULT_CAPTURE_NONTENSOR_STACK
2217
+ elif _CAPTURE_NONTENSOR_STACK == "none":
2218
+ return _DEFAULT_CAPTURE_NONTENSOR_STACK
2219
return (
2220
strtobool(_CAPTURE_NONTENSOR_STACK)
2221
if isinstance(_CAPTURE_NONTENSOR_STACK, str)
0 commit comments