Skip to content

Commit 6038c7d

Browse files
author
Vincent Moens
committed
amend
1 parent 12c2717 commit 6038c7d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

torchrl/data/tensor_specs.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@
3838
TensorDictBase,
3939
unravel_key,
4040
)
41+
from tensordict.base import NO_DEFAULT
4142
from tensordict.utils import _getitem_batch_size, NestedKey
4243
from torchrl._utils import _make_ordinal_device, get_binary_env_var
4344

@@ -79,8 +80,6 @@
7980
" an issue at https://github.com/pytorch/rl/issues"
8081
)
8182

82-
NO_DEFAULT = object()
83-
8483

8584
def _default_dtype_and_device(
8685
dtype: Union[None, torch.dtype],

0 commit comments

Comments
 (0)