[SOT][3.12] remove distutils import to avoid import error (PaddlePadd…

…le#58117)
danleifeng · Oct 17, 2023 · 9bf67db · 9bf67db
1 parent 103b6fa
commit 9bf67db
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/paddle/jit/sot/opcode_translator/skip_files.py b/python/paddle/jit/sot/opcode_translator/skip_files.py
@@ -19,7 +19,6 @@
 import copy
 import copyreg
 import dataclasses
-import distutils
 import enum
 import functools
 import importlib
@@ -94,7 +93,6 @@
     codecs,
     uuid,
     setuptools,
-    distutils,
     warnings,
 }
 
@@ -105,6 +103,11 @@
     NEED_SKIP_THIRD_PARTIY_MODULES.add(sre_compile)
     NEED_SKIP_THIRD_PARTIY_MODULES.add(sre_parse)
 
+if sys.version_info < (3, 12):
+    import distutils
+
+    NEED_SKIP_THIRD_PARTIY_MODULES.add(distutils)
+
 
 def _strip_init_py(s):
     return re.sub(r"__init__.py$", "", s)