Skip to content

Commit ca6bf32

Browse files
committed
Fix test
1 parent 65eccfe commit ca6bf32

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

extension/llm/export/test/test_export_llm.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -97,12 +97,8 @@ def test_with_config(self, mock_export_llama: MagicMock) -> None:
9797
self.assertEqual(
9898
called_config.quantization.pt2e_quantize.value, "xnnpack_dynamic"
9999
)
100-
self.assertEqual(
101-
called_config.quantization.use_spin_quant.value, "cuda"
102-
)
103-
self.assertEqual(
104-
called_config.backend.coreml.quantize.value, "c4w"
105-
)
100+
self.assertEqual(called_config.quantization.use_spin_quant.value, "cuda")
101+
self.assertEqual(called_config.backend.coreml.quantize.value, "c4w")
106102
self.assertEqual(
107103
called_config.backend.coreml.compute_units.value, "cpu_and_gpu"
108104
)

0 commit comments

Comments
 (0)