[Easy][pytree] enable CXX pytree under torch::deploy (pytorch#130144)

XuehaiPan · DiweiSun · commit 453db63219e1 · 2024-07-21T20:40:36.000-07:00
Pull Request resolved: pytorch#130144 Approved by: https://github.com/zou3519 ghstack dependencies: pytorch#130895, pytorch#130139
diff --git a/torch/utils/_cxx_pytree.py b/torch/utils/_cxx_pytree.py
@@ -29,11 +29,6 @@
 )
 from typing_extensions import deprecated
 
-import torch
-
-if torch._running_with_deploy():  # type: ignore[no-untyped-call]
-    raise ImportError("C++ pytree utilities do not work with torch::deploy.")
-
 import optree
 from optree import PyTreeSpec  # direct import for type annotations
 
