Update

[ghstack-poisoned]
pytorch · XuehaiPan · Jun 24, 2024 · Jun 24, 2024 · Jun 24, 2024 · Jun 24, 2024
commit b6b813937f9522c50225a4b435a901b74a194f75
diff --git a/torchgen/gen.py b/torchgen/gen.py
@@ -4,7 +4,6 @@
 import functools
 import json
 import os
-import pathlib
 from collections import defaultdict, namedtuple, OrderedDict
 from dataclasses import dataclass, field
 from pathlib import Path
@@ -2706,12 +2705,12 @@ def gen_declarations_yaml(
     )
 
 
-def get_torchgen_root() -> pathlib.Path:
+def get_torchgen_root() -> Path:
     """
     If you're depending on torchgen out-of-tree, you can use the root to figure
     out the path to native_functions.yaml
     """
-    return pathlib.Path(__file__).parent.resolve()
+    return Path(__file__).parent.resolve()
 
 
 def main() -> None:
@@ -2873,11 +2872,11 @@ def main() -> None:
     #
     #   Invalid character escape '\c'.
     core_install_dir = f"{options.install_dir}/core"
-    pathlib.Path(core_install_dir).mkdir(parents=True, exist_ok=True)
+    Path(core_install_dir).mkdir(parents=True, exist_ok=True)
     ops_install_dir = f"{options.install_dir}/ops"
-    pathlib.Path(ops_install_dir).mkdir(parents=True, exist_ok=True)
+    Path(ops_install_dir).mkdir(parents=True, exist_ok=True)
     aoti_install_dir = f"{options.aoti_install_dir}"
-    pathlib.Path(aoti_install_dir).mkdir(parents=True, exist_ok=True)
+    Path(aoti_install_dir).mkdir(parents=True, exist_ok=True)
 
     core_fm = make_file_manager(options=options, install_dir=core_install_dir)
     cpu_fm = make_file_manager(options=options)
@@ -2964,7 +2963,7 @@ def main() -> None:
         gen_declarations_yaml(native_functions=native_functions, cpu_fm=cpu_fm)
 
     if options.output_dependencies:
-        depfile_path = pathlib.Path(options.output_dependencies).resolve()
+        depfile_path = Path(options.output_dependencies).resolve()
         depfile_name = depfile_path.name
         depfile_stem = depfile_path.stem
 

diff --git a/torchgen/gen_executorch.py b/torchgen/gen_executorch.py
@@ -2,7 +2,6 @@
 
 import argparse
 import os
-import pathlib
 from collections import defaultdict
 from dataclasses import dataclass
 from pathlib import Path
@@ -984,7 +983,7 @@ def main() -> None:
             )
 
     if options.output_dependencies:
-        depfile_path = pathlib.Path(options.output_dependencies).resolve()
+        depfile_path = Path(options.output_dependencies).resolve()
         depfile_name = depfile_path.name
         depfile_stem = depfile_path.stem