Add xfail with sys.platform == "darwin" for mac

shark/tests/test_shark_importer.py

@@ -5,6 +5,7 @@ import pytest
 from shark.parser import shark_args
 from shark.shark_inference import SharkInference
 from shark.tflite_utils import TFLitePreprocessor
+import sys
 
 # model_path = "https://tfhub.dev/tensorflow/lite-model/albert_lite_base/squadv1/1?lite-format=tflite"
 
@@ -121,7 +122,7 @@ pytest_param = pytest.mark.parametrize(
 
 
 @pytest_param
-@pytest.mark.xfail(reason="known macos tflite install issue")
+@pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
 def test_albert(dynamic, device):
     module_tester = AlbertTfliteModuleTester(dynamic=dynamic, device=device)
     module_tester.create_and_check_module()

tank/albert_lite_base/albert_lite_base_tflite_mlir_test.py

@@ -1,3 +1,5 @@
+import sys
+
 from shark.shark_downloader import SharkDownloader
 from shark.shark_inference import SharkInference
 import pytest
@@ -53,7 +55,9 @@ class AlbertTfliteModuleTest(unittest.TestCase):
         self.module_tester = AlbertTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/albert_lite_base/albert_lite_base_tflite_test.py

@@ -124,7 +124,9 @@ class AlbertTfliteModuleTest(unittest.TestCase):
         self.module_tester = AlbertTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/arbitrary-image-stylization-v1-256/arbitrary-image-stylization-v1-256_tflite_test.py

@@ -87,7 +87,9 @@ class ArbitraryImageStylizationV1TfliteModuleTest(unittest.TestCase):
         self.module_tester = ArbitraryImageStylizationV1TfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/birds_V1/birds_V1_tflite_test.py

@@ -121,7 +121,9 @@ class BirdsV1TfliteModuleTest(unittest.TestCase):
         self.module_tester = BirdsV1TfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/cartoongan/cartoongan_tflite_test.py

@@ -86,7 +86,9 @@ class CartoonganTfliteModuleTest(unittest.TestCase):
         self.module_tester = CartoonganTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/deeplabv3/deeplabv3_tflite_test.py

@@ -126,7 +126,9 @@ class DeepLabV3TfliteModuleTest(unittest.TestCase):
         self.module_tester = DeepLabV3TfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/densenet/densenet_tflite_test.py

@@ -89,7 +89,9 @@ class DensenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = DensenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/efficientnet_224_fp32/efficientnet_224_fp32_tflite_test.py

@@ -120,7 +120,9 @@ class Efficientnet_224_fp32TfliteModuleTest(unittest.TestCase):
         self.module_tester = Efficientnet_224_fp32TfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/efficientnet_lite0_fp32_2/efficientnet_lite0_fp32_2_tflite_test.py

@@ -120,7 +120,9 @@ class Efficientnet_lite0_fp32_2TfliteModuleTest(unittest.TestCase):
         self.module_tester = Efficientnet_lite0_fp32_2TfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/efficientnet_lite0_int8_2/efficientnet_lite0_int8_2_tflite_test.py

@@ -118,7 +118,9 @@ class Efficientnet_lite0_int8_2TfliteModuleTest(unittest.TestCase):
         self.module_tester = Efficientnet_lite0_int8_2TfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/inception_v4_299_fp32/inception_v4_299_fp32_tflite_test.py

@@ -120,7 +120,9 @@ class Inception_v4_299_fp32TfliteModuleTest(unittest.TestCase):
         self.module_tester = Inception_v4_299_fp32TfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/inception_v4_299_uint8/inception_v4_299_uint8_tflite_test.py

@@ -117,7 +117,9 @@ class Inception_v4_299_uint8TfliteModuleTest(unittest.TestCase):
         self.module_tester = Inception_v4_299_uint8TfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/midas/midas_tflite_test.py

@@ -89,7 +89,9 @@ class MidasTfliteModuleTest(unittest.TestCase):
         self.module_tester = MidasTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mnasnet_1.0_224/mnasnet_tflite_test.py

@@ -89,7 +89,9 @@ class MnasnetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MnasnetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilebert-baseline-tf2-float/mobilebert-baseline-tf2-float_tflite_test.py

@@ -118,7 +118,9 @@ class MobilebertTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilebertTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilebert-baseline-tf2-quant/mobilebert-baseline-tf2-quant_tflite_test.py

@@ -118,7 +118,9 @@ class MobilebertTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilebertTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilebert-edgetpu-s-float/mobilebert-edgetpu-s-float_tflite_test.py

@@ -124,7 +124,9 @@ class MobilebertTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilebertTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilebert-edgetpu-s-quant/mobilebert-edgetpu-s-quant_tflite_test.py

@@ -106,7 +106,9 @@ class MobilebertTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilebertTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilebert/mobilebert_tflite_test.py

@@ -117,7 +117,9 @@ class MobilebertTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilebertTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilenet_v1_224_1.0_float/mobilenet_v1_float_tflite_test.py

@@ -119,7 +119,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilenet_v1_224_1.0_uint8/mobilenet_v1_uint8_tflite_test.py

@@ -117,7 +117,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilenet_v2_1.00_224_int8/mobilenet_v2_int8_tflite_test.py

@@ -119,7 +119,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilenet_v2_1.0_224/mobilenet_v2_tflite_test.py

@@ -120,7 +120,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilenet_v2_224_1.0_uint8/mobilenet_v2_uint8_tflite_test.py

@@ -117,7 +117,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilenet_v3-large_224_1.0_float/mobilenet_v3_float_tflite_test.py

@@ -120,7 +120,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilenet_v3-large_224_1.0_uint8/mobilenet_v3_tflite_test.py

@@ -117,7 +117,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/mobilenet_v3.5multiavg_1.00_224_int8/mobilenet_v35_int8_tflite_test.py

@@ -120,7 +120,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/multi_person_mobilenet_v1_075_float/multi_person_mobilenet_tflite_test.py

@@ -86,7 +86,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/resnet_50_224_int8/resnet_50_224_int8_tflite_test.py

@@ -89,7 +89,9 @@ class ResnetTfliteModuleTest(unittest.TestCase):
         self.module_tester = ResnetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/squeezenet/squeezenet_tflite_test.py

@@ -89,7 +89,9 @@ class SequeezeNetTfliteModuleTest(unittest.TestCase):
         self.module_tester = SequeezeNetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/ssd_mobilenet_v1_320_1.0_float/ssd_mobilenet_v1_float_tflite_test.py

@@ -117,7 +117,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/ssd_mobilenet_v1_320_1.0_uint8/ssd_mobilenet_v1_uint8_tflite_test.py

@@ -114,7 +114,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/ssd_mobilenet_v2_face_quant/ssd_mobilenet_v2_face_quant_tflite_test.py

@@ -124,7 +124,9 @@ class MobilenetTfliteModuleTest(unittest.TestCase):
         self.module_tester = MobilenetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/ssd_spaghettinet_edgetpu_large/ssd_spaghettinet_edgetpu_large_tflite_test.py

@@ -116,7 +116,9 @@ class SpaghettinetTfliteModuleTest(unittest.TestCase):
         self.module_tester = SpaghettinetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"

tank/ssd_spaghettinet_edgetpu_large_uint8/ssd_spaghettinet_edgetpu_large_uint8_tflite_test.py

@@ -114,7 +114,9 @@ class SpaghettinetTfliteModuleTest(unittest.TestCase):
         self.module_tester = SpaghettinetTfliteModuleTester(self)
         self.module_tester.save_mlir = self.save_mlir
 
-    @pytest.mark.xfail(reason="known macos tflite install issue")
+    import sys
+
+    @pytest.mark.xfail(sys.platform == "darwin", reason="known macos tflite install issue")
     def test_module_static_cpu(self):
         self.module_tester.dynamic = False
         self.module_tester.device = "cpu"