From 04cbb4191c74879b98db82eff3df1e26cb0abc25 Mon Sep 17 00:00:00 2001 From: wrmsr Date: Mon, 1 Sep 2025 11:50:20 -0700 Subject: [PATCH] Fix dequantize python sig (#2562) --- python/src/ops.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/src/ops.cpp b/python/src/ops.cpp index 7b585af6e..74909e81d 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -4271,7 +4271,7 @@ void init_ops(nb::module_& m) { nb::kw_only(), "stream"_a = nb::none(), nb::sig( - "def dequantize(w: array, /, scales: array, biases: Optional[array] = = None, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, stream: Union[None, Stream, Device] = None) -> array"), + "def dequantize(w: array, /, scales: array, biases: Optional[array] = None, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, stream: Union[None, Stream, Device] = None) -> array"), R"pbdoc( Dequantize the matrix ``w`` using quantization parameters.