Replace get_platform with sys.platform and platform.machine

PyO3 · Nov 10, 2021 · c1a303f · c1a303f
1 parent 9a22e33
commit c1a303f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -19,7 +19,6 @@
 import toml
 from setuptools import setup
 from setuptools.command.install import install
-from setuptools._distutils.util import get_platform
 
 # Force the wheel to be platform specific
 # https://stackoverflow.com/a/45150383/3549270
@@ -70,9 +69,8 @@ def run(self):
                 "--message-format=json",
             ]
 
-            if (
-                platform.machine() in ("ppc64le", "ppc64", "powerpc")
-                or get_platform() == "win-arm64"
+            if platform.machine() in ("ppc64le", "ppc64", "powerpc") or (
+                sys.platform == "win32" and platform.machine() == "ARM64"
             ):
                 cargo_args.extend(
                     ["--no-default-features", "--features=upload,log,human-panic"]