chore: Make sure to mark builds as platform specific

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Mar 22, 2024 · ae453fc · ae453fc
1 parent e9ef3ca
commit ae453fc
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -1,3 +1,5 @@
+# type: ignore
+
 import glob
 import os
 import platform
@@ -283,7 +285,7 @@ def initialize_options(self):
 
     def finalize_options(self):
         develop.finalize_options(self)
-        if NO_TS:
+        if NO_TS or PY_ONLY:
             self.root_is_pure = False
 
     def run(self):
@@ -307,7 +309,7 @@ def initialize_options(self):
 
     def finalize_options(self):
         install.finalize_options(self)
-        if NO_TS:
+        if NO_TS or PY_ONLY:
             self.root_is_pure = False
 
     def run(self):
@@ -331,7 +333,7 @@ def initialize_options(self):
 
     def finalize_options(self):
         bdist_wheel.finalize_options(self)
-        if NO_TS:
+        if NO_TS or PY_ONLY:
             self.root_is_pure = False
 
     def run(self):
@@ -354,7 +356,7 @@ def initialize_options(self):
 
     def finalize_options(self):
         editable_wheel.finalize_options(self)
-        if NO_TS:
+        if NO_TS or PY_ONLY:
             self.root_is_pure = False
 
     def run(self):