Skip windows

Xilinx · Nov 13, 2024 · df607eb · df607eb
1 parent ff6ea28
commit df607eb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/brevitas_examples/test_llm.py b/tests/brevitas_examples/test_llm.py
@@ -417,9 +417,8 @@ def test_small_models_quant_layer(caplog, layer_args):
     if args.replace_rmsnorm:
         if torch_version < version.parse('2.4'):
             pytest.skip("Replacing RMSNorm requires torch 2.4+ or greater")
-        if hasattr(
-                args,
-                'graph_rotation') and args.graph_rotation == 'fx' and platform.system() == 'win32':
+        if hasattr(args, 'graph_rotation') and args.graph_rotation == 'fx' and platform.system(
+        ) == 'Windows':
             pytest.skip("Skipping dynamo + windows")
     float_ppl, quant_ppl, model = validate_args_and_run_main(args)
     assert_layer_types(model, exp_layer_types)