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

XuehaiPan · pytorchmergebot · commit 1439bd3c9c0a · 2024-07-21T07:36:22.000Z
Pull Request resolved: #130144 Approved by: https://github.com/zou3519 ghstack dependencies: #130895, #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
 
