Skip to content

Commit 44fdc96

Browse files
committed
Fix lint check issue.
Signed-off-by: menogrey <1299267905@qq.com>
1 parent 4faf918 commit 44fdc96

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/ut/quantization/test_quant_config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from typing import Any
12
from unittest.mock import MagicMock, patch
23

34
import torch
@@ -64,7 +65,7 @@ def test_from_config(self):
6465
def test_override_quantization_method(self, mock_is_available):
6566
# Test when NPU is available
6667
mock_is_available.return_value = True
67-
hf_quant_cfg = {}
68+
hf_quant_cfg: dict[str, Any] = {}
6869
result = AscendQuantConfig.override_quantization_method(hf_quant_cfg, None)
6970
self.assertEqual(result, ASCEND_QUANTIZATION_METHOD)
7071

0 commit comments

Comments
 (0)