We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4faf918 commit 44fdc96Copy full SHA for 44fdc96
tests/ut/quantization/test_quant_config.py
@@ -1,3 +1,4 @@
1
+from typing import Any
2
from unittest.mock import MagicMock, patch
3
4
import torch
@@ -64,7 +65,7 @@ def test_from_config(self):
64
65
def test_override_quantization_method(self, mock_is_available):
66
# Test when NPU is available
67
mock_is_available.return_value = True
- hf_quant_cfg = {}
68
+ hf_quant_cfg: dict[str, Any] = {}
69
result = AscendQuantConfig.override_quantization_method(hf_quant_cfg, None)
70
self.assertEqual(result, ASCEND_QUANTIZATION_METHOD)
71
0 commit comments