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 a2b7a48 commit 4ac95ceCopy full SHA for 4ac95ce
onnxruntime/python/tools/quantization/base_quantizer.py
@@ -42,7 +42,7 @@ def __init__(self, **data: dict[str, Any]):
42
for k, v in data.items():
43
if not isinstance(k, str):
44
raise TypeError(f"Keys must be strings not {type(k)} for k={k!r}.")
45
- if k != "axis" and not isinstance(v, (int, str, np.ndarray)):
+ if k != "axis" and not isinstance(v, (int, str, np.ndarray, float)):
46
raise TypeError(f"Values must be numpy arrays, int, float, str not {type(v)} for k={k!r}.")
47
if k == "axis" and not isinstance(v, int) and v is not None:
48
raise TypeError(f"Axis value must be an int or None, not {type(v)}.")
0 commit comments