Skip to content

Commit 2f60f8d

Browse files
author
Vincent Moens
committed
[BugFix] none value for _CAPTURE_NONTENSOR_STACK
ghstack-source-id: 0d38f48 Pull Request resolved: #1241
1 parent 539651f commit 2f60f8d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tensordict/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2214,6 +2214,8 @@ def capture_non_tensor_stack(allow_none=False):
22142214
return None
22152215
elif _CAPTURE_NONTENSOR_STACK is None:
22162216
return _DEFAULT_CAPTURE_NONTENSOR_STACK
2217+
elif _CAPTURE_NONTENSOR_STACK == "none":
2218+
return _DEFAULT_CAPTURE_NONTENSOR_STACK
22172219
return (
22182220
strtobool(_CAPTURE_NONTENSOR_STACK)
22192221
if isinstance(_CAPTURE_NONTENSOR_STACK, str)

0 commit comments

Comments
 (0)