f

microsoft · Jan 12, 2024 · 01bfa24 · 01bfa24
1 parent 0448cd5
commit 01bfa24
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/onnxruntime/python/tools/quantization/matmul_4bits_quantizer.py b/onnxruntime/python/tools/quantization/matmul_4bits_quantizer.py
@@ -11,7 +11,6 @@
 import importlib
 import logging
 import os
-from typing import Optional
 
 import numpy as np
 import numpy.typing as npt
@@ -157,7 +156,7 @@ def optimize_weights(
         zero: np.ndarray,
         min_max: list[int],
         axis: int = 0,
-        opt_params: Optional[dict] = None,
+        opt_params: dict = None,  # noqa: RUF013
         verbose=False,
     ):
         opt_params = {"lp_norm": 0.7, "beta": 1e1, "kappa": 1.01, "iters": 20} if opt_params is None else opt_params