amend

pytorch · Jul 8, 2024 · ca747d4 · ca747d4
1 parent 7e8c1bd
commit ca747d4
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 7 deletions.
diff --git a/tensordict/__init__.py b/tensordict/__init__.py
@@ -26,11 +26,11 @@
     set_lazy_legacy,
 )
 from tensordict._pytree import *
-from tensordict._C import unravel_key, unravel_key_list  # ignore
+from tensordict._C import unravel_key, unravel_key_list  # @manual=//tensordict:_C
 from tensordict.nn import TensorDictParams
 
 try:
-    from tensordict.version import __version__
+    from tensordict.version import __version__  # @manual=//tensordict:version
 except ImportError:
     __version__ = None
 

diff --git a/tensordict/_lazy.py b/tensordict/_lazy.py
@@ -45,7 +45,7 @@
     from tensordict.utils import _ftdim_mock as ftdim
 
     _has_funcdim = False
-from tensordict._C import _unravel_key_to_tuple, unravel_key_list  # ignore
+from tensordict._C import _unravel_key_to_tuple, unravel_key_list  # @manual=//tensordict:_C
 from tensordict._td import _SubTensorDict, _TensorDictKeysView, TensorDict
 from tensordict.base import (
     _is_leaf_nontensor,

diff --git a/tensordict/nn/common.py b/tensordict/nn/common.py
@@ -13,7 +13,7 @@
 
 import torch
 from cloudpickle import dumps as cloudpickle_dumps, loads as cloudpickle_loads
-from tensordict._C import _unravel_key_to_tuple, unravel_key_list  # ignore
+from tensordict._C import _unravel_key_to_tuple, unravel_key_list  # @manual=//tensordict:_C
 
 from tensordict._td import is_tensor_collection, TensorDictBase
 from tensordict.functional import make_tensordict

diff --git a/tensordict/nn/sequence.py b/tensordict/nn/sequence.py
@@ -26,7 +26,7 @@
     FUNCTORCH_ERROR = "functorch not installed. Consider installing functorch to use this functionality."
 
 
-from tensordict._C import unravel_key_list  # ignore
+from tensordict._C import unravel_key_list  # @manual=//tensordict:_C
 from tensordict.nn.common import dispatch, TensorDictModule
 from tensordict.tensordict import LazyStackedTensorDict, TensorDictBase
 from tensordict.utils import NestedKey

diff --git a/tensordict/tensorclass.py b/tensordict/tensorclass.py
@@ -30,7 +30,7 @@
 import tensordict as tensordict_lib
 
 import torch
-from tensordict._C import _unravel_key_to_tuple  # ignore
+from tensordict._C import _unravel_key_to_tuple  # @manual=//tensordict:_C
 from tensordict._lazy import LazyStackedTensorDict
 from tensordict._pytree import _register_td_node
 from tensordict._td import is_tensor_collection, NO_DEFAULT, TensorDict, TensorDictBase

diff --git a/tensordict/utils.py b/tensordict/utils.py
@@ -46,7 +46,7 @@
 except ImportError:
     _has_funcdim = False
 from packaging.version import parse
-from tensordict._C import (  # noqa: F401  # ignore
+from tensordict._C import (  # noqa: F401  # @manual=//tensordict:_C
     _unravel_key_to_tuple,
     unravel_key,
     unravel_key_list,