BUG: fix a regression where np.linspace's num argument would be ignor…

…ed for unyt_arrays
yt-project · Dec 25, 2024 · c25ed28 · c25ed28
1 parent 41b4b36
commit c25ed28
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/unyt/_array_functions.py b/unyt/_array_functions.py
@@ -666,7 +666,7 @@ def linspace(
         return _linspace(
             start,
             stop,
-            num=50,
+            num=num,
             endpoint=endpoint,
             retstep=retstep,
             dtype=dtype,
@@ -681,7 +681,7 @@ def linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis
         return _linspace(
             start,
             stop,
-            num=50,
+            num=num,
             endpoint=endpoint,
             retstep=retstep,
             dtype=dtype,