From ca747d47b5560d5ffb48981c40022fe633b11ab0 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Mon, 8 Jul 2024 16:09:58 +0100
Subject: [PATCH] amend

---
 tensordict/__init__.py    | 4 ++--
 tensordict/_lazy.py       | 2 +-
 tensordict/nn/common.py   | 2 +-
 tensordict/nn/sequence.py | 2 +-
 tensordict/tensorclass.py | 2 +-
 tensordict/utils.py       | 2 +-
 6 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/tensordict/__init__.py b/tensordict/__init__.py
index 93abe9e49..7640bde08 100644
--- 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
index 8716091c8..915a55353 100644
--- 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
index 2619ded58..cf36a31b5 100644
--- 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
index 15bbce9d8..b4230b5e4 100644
--- 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
index 4750765ca..ca7b1973f 100644
--- 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
index fc03fbb7c..e0d894fb2 100644
--- 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,