Replace imp package with importlib

Luos-io · Dec 1, 2023 · 0eb8361 · 0eb8361
1 parent 03d4182
commit 0eb8361
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -1,11 +1,14 @@
 #!/usr/bin/env python
 
-import imp
+try:
+    import imp
+    version = imp.load_source('pyluos.version', 'pyluos/version.py')
+except ImportError:
+    from importlib.machinery import SourceFileLoader
+    version = SourceFileLoader('pyluos.version', 'pyluos/version.py').load_module()
 
 from setuptools import setup, find_packages
 
-version = imp.load_source('pyluos.version', 'pyluos/version.py')
-
 with open("README.md", "r") as fh:
     long_description = fh.read()