Add platform checks before nested pip call (#3084)

* Add platform checks before nested pip call

Signed-off-by: ddelange <[email protected]>

* Remove unused variables

Signed-off-by: ddelange <[email protected]>

* PR Suggestion

Signed-off-by: ddelange <[email protected]>

---------

Signed-off-by: ddelange <[email protected]>

Author: ddelange

python/packaging/frontend_sdist/setup.py

@@ -16,6 +16,7 @@
 #
 
 import os
+import platform
 import subprocess
 import sys
 
@@ -39,6 +40,15 @@ def run_pip_command(args, call_func):
         return call_func([os.path.join(sys.exec_prefix, "bin", "pip")] + args)
 
 
+# check wheel availability using information from https://github.com/pypa/packaging/blob/23.1/src/packaging/markers.py#L175-L190
+if sys.platform not in ("linux", "win32"):
+    raise RuntimeError("TensorRT currently only builds wheels for Linux and Windows")
+if sys.implementation.name != "cpython":
+    raise RuntimeError("TensorRT currently only builds wheels for CPython")
+if platform.machine() not in ("x86_64", "AMD64"):
+    raise RuntimeError("TensorRT currently only builds wheels for x86_64 processors")
+
+
 class InstallCommand(install):
     def run(self):
         # pip-inside-pip hack ref #3080