.

Lightning-AI · Feb 7, 2021 · cb8dc27 · cb8dc27
1 parent 595044b
commit cb8dc27
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/callbacks/test_quantization.py b/tests/callbacks/test_quantization.py
@@ -23,7 +23,8 @@
 from tests.helpers.zoo_data_models import RegressDataModule, RegressionModel
 
 
-@pytest.mark.parametrize("observe,fuse", [(None, True), ('histogram', False), ('average', True)])
+@pytest.mark.parametrize("observe", ['histogram', 'average'])
+@pytest.mark.parametrize("fuse", [True, False])
 def test_quantization(tmpdir, observe, fuse):
     """Parity for  quant model"""
     seed_everything(42)