Update

[ghstack-poisoned]
pytorch · Oct 29, 2024 · 1eaac7e · 1eaac7e
1 parent fb38803
commit 1eaac7e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensordict/_td.py b/tensordict/_td.py
@@ -121,7 +121,7 @@
             is_batchedtensor,
         )
     except ImportError:
-        from functorch._C import is_batchedtensor  # @manual=fbcode//functorch:_C
+        from functorch._C import is_batchedtensor  # @manual=fbcode//caffe2/functorch:_C
 
     _has_functorch = True
 except ImportError:

diff --git a/tensordict/utils.py b/tensordict/utils.py
@@ -82,7 +82,7 @@
             is_batchedtensor,
         )
     except ImportError:
-        from functorch._C import (  # @manual=fbcode//functorch:_C  # noqa
+        from functorch._C import (  # @manual=fbcode//caffe2/functorch:_C  # noqa
             get_unwrapped,
             is_batchedtensor,
         )