amend

pytorch · Jul 9, 2024 · 3dc583d · 3dc583d
1 parent 2d2b68f
commit 3dc583d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensordict/utils.py b/tensordict/utils.py
@@ -74,10 +74,10 @@
             is_batchedtensor,
         )
     except ImportError:
-        from functorch._C import (
+        from functorch._C import (  # @manual=fbcode//functorch:_C  # noqa
             get_unwrapped,
             is_batchedtensor,
-        )  # @manual=fbcode//functorch:_C
+        )
 except ImportError:
     pass