[Fix] Fix AssertionError when importing pip before setuptools (#198)

* fix AssertionError when importing pip before setuptools * refine comments
open-mmlab · Feb 4, 2023 · 959f9c9 · 959f9c9
1 parent 213f1b2
commit 959f9c9
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/mim/__init__.py b/mim/__init__.py
@@ -1,4 +1,18 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+
+# NOTE: We could got AssertionError when importing pip before
+# setuptools. A workaround is to import setuptools first and filter
+# warnings that are caused by setuptools replacing distutils.
+# Related issues:
+# - https://github.com/pypa/setuptools/issues/3621
+# - https://github.com/open-mmlab/mmclassification/issues/1343
+try:
+    import setuptools  # noqa: F401
+    import warnings
+    warnings.filterwarnings('ignore', 'Setuptools is replacing distutils')
+except ImportError:
+    pass
+
 from .commands import (
     download,
     get_model_info,