Fix (test): narrowing test cases

Xilinx · Mar 20, 2024 · ed03c0e · ed03c0e
1 parent 730aef0
commit ed03c0e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/brevitas/graph/test_equalization.py b/tests/brevitas/graph/test_equalization.py
@@ -60,9 +60,8 @@ def test_resnet18_equalization():
 
 
 @pytest_cases.parametrize("merge_bias", [True, False])
-@pytest_cases.parametrize("scale_computation_type", ["maxabs", "range", "l1"])
 def test_equalization_torchvision_models(
-        model_coverage: tuple, merge_bias: bool, scale_computation_type: str):
+        model_coverage: tuple, merge_bias: bool, scale_computation_type: str = "maxabs"):
     model, coverage = model_coverage
 
     torch.manual_seed(SEED)