Correct test skipping

Xilinx · Jan 27, 2025 · efd6dbd · efd6dbd
1 parent aaf74bb
commit efd6dbd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/brevitas/graph/test_quantize.py b/tests/brevitas/graph/test_quantize.py
@@ -65,7 +65,6 @@ def test_layerwise_quantize_blacklist(kwargs):
                         K: torch.matmul(tensor, rot_mat),
             'key': '0',
             'expected': "<class 'torch.nn.utils.parametrize.ParametrizedQuantLinear'>"},])
-@requires_pt_lt('2.2.2', 'Windows')
 def test_layerwise_quantize_parametrized_modules(kwargs):
     key = kwargs['key']
     exp = kwargs['expected']
@@ -162,6 +161,8 @@ def test_remove_parametrization_entries_state_dict(kwargs):
             'key': '0',
             'expected': "<class 'torch.nn.utils.parametrize.ParametrizedQuantLinear'>"},])
 def test_quantize_parametrized_modules(kwargs):
+    if platform.system() == "Windows":
+        pytest.skip("Skipping dynamo + windows")
     key = kwargs['key']
     exp = kwargs['expected']
     rot_mat = kwargs['rot_mat']

diff --git a/tests/brevitas_examples/test_llm.py b/tests/brevitas_examples/test_llm.py
@@ -979,8 +979,8 @@ def test_small_models_rotation_optimization_ppl(
 @requires_pt_ge('2.4')
 def test_small_models_rotation_optimization_layer_count(
         caplog, rotation_optimization_args_layer_count_and_ppl):
-    if platform.system() == "Windows":
-        pytest.skip("Skipping dynamo + windows")
+    if platform.system() != "Linux":
+        pytest.skip("Skipping dynamo + windows/macos")
     # Tolerances are stricter for this test, to ensure that it does not pass
     # with non-optimized quantized perplexities
     caplog.set_level(logging.INFO)