amend

pytorch · Aug 5, 2024 · 6038c7d · 6038c7d
1 parent 12c2717
commit 6038c7d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torchrl/data/tensor_specs.py b/torchrl/data/tensor_specs.py
@@ -38,6 +38,7 @@
     TensorDictBase,
     unravel_key,
 )
+from tensordict.base import NO_DEFAULT
 from tensordict.utils import _getitem_batch_size, NestedKey
 from torchrl._utils import _make_ordinal_device, get_binary_env_var
 
@@ -79,8 +80,6 @@
     " an issue at https://github.com/pytorch/rl/issues"
 )
 
-NO_DEFAULT = object()
-
 
 def _default_dtype_and_device(
     dtype: Union[None, torch.dtype],