From 9320eb89a8a44b3d2edfdf23cced2cd075fbf580 Mon Sep 17 00:00:00 2001 From: wrmsr Date: Tue, 11 Nov 2025 09:55:24 -0800 Subject: [PATCH] Fix dequantize python sig (dtype default) (#2752) --- 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 9816837ba..c28cf6a51 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -4335,7 +4335,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: Optional[int] = None, bits: Optional[int] = None, mode: str = 'affine', dtype: Optional[Dtype], *, stream: Union[None, Stream, Device] = None) -> array"), + "def dequantize(w: array, /, scales: array, biases: Optional[array] = None, group_size: Optional[int] = None, bits: Optional[int] = None, mode: str = 'affine', dtype: Optional[Dtype] = None, *, stream: Union[None, Stream, Device] = None) -> array"), R"pbdoc( Dequantize the matrix ``w`` using quantization parameters.