Update test_torch_quantization_ops.py

metascroy · web-flow · commit 30b74a15f3f3 · 2025-09-05T09:09:12.000-07:00
diff --git a/coremltools/converters/mil/frontend/torch/test/test_torch_quantization_ops.py b/coremltools/converters/mil/frontend/torch/test/test_torch_quantization_ops.py
@@ -275,7 +275,7 @@ def forward(self, x):
     @pytest.mark.skipif(not _HAS_TORCHAO, reason=MSG_TORCHAO_NOT_FOUND)
     @pytest.mark.parametrize(
         "compute_unit, has_zeros",
-        itertools.product(compute_units, [True, False], [ct.target.IOS16, ct.target.IOS17]),
+        itertools.product(compute_units, [True, False], [ct.target.iOS16, ct.target.iOS17]),
     )
     def test_dequantize_affine_before_ios18(self, compute_unit, has_zeros, minimum_deployment_target):
 

Original file line number	Diff line number	Diff line change
`@@ -275,7 +275,7 @@ def forward(self, x):`
`275`	`275`	`@pytest.mark.skipif(not _HAS_TORCHAO, reason=MSG_TORCHAO_NOT_FOUND)`
`276`	`276`	`@pytest.mark.parametrize(`
`277`	`277`	`"compute_unit, has_zeros",`
`278`		`- itertools.product(compute_units, [True, False], [ct.target.IOS16, ct.target.IOS17]),`
	`278`	`+ itertools.product(compute_units, [True, False], [ct.target.iOS16, ct.target.iOS17]),`
`279`	`279`	`)`
`280`	`280`	`def test_dequantize_affine_before_ios18(self, compute_unit, has_zeros, minimum_deployment_target):`
`281`	`281`