From 220fec438dbc3dbaf3879288eb0f4c6586136808 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Tue, 29 Oct 2024 10:50:49 +0000
Subject: [PATCH] [Minor] Fix wrong fbcode manual links

ghstack-source-id: 8ff9fb4a98ba3cbfc7c90d12a963a8a1e950e95b
Pull Request resolved: https://github.com/pytorch/tensordict/pull/1064

(cherry picked from commit b06de95a901646e3d7ed29a03543f8523428d738)
---
 tensordict/_td.py   | 2 +-
 tensordict/utils.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensordict/_td.py b/tensordict/_td.py
index b41f0947a..a8f3ddce7 100644
--- 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
index 65f0ef4c4..81f0ecb00 100644
--- 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,
         )