Fix mx.core.linspace type annotation (#2820)

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
2025-11-24 23:15:08 +01:00
parent 8abd37ad05
commit 1d4eacb737
1 changed files with 1 additions and 1 deletions
@@ -1445,7 +1445,7 @@ void init_ops(nb::module_& m) {
      "dtype"_a.none() = mx::float32,
      "stream"_a = nb::none(),
      nb::sig(
-          "def linspace(start, stop, num: Optional[int] = 50, dtype: Optional[Dtype] = float32, stream: Union[None, Stream, Device] = None) -> array"),
+          "def linspace(start: scalar, stop: scalar, num: Optional[int] = 50, dtype: Optional[Dtype] = float32, stream: Union[None, Stream, Device] = None) -> array"),
      R"pbdoc(
        Generate ``num`` evenly spaced numbers over interval ``[start, stop]``.