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

…le#58117)
hitywt · Oct 24, 2023 · ba5b95f · ba5b95f
1 parent c87a3f4
commit ba5b95f
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)