Merge pull request #553 from neutrinoceros/bug/fix_linspace_num

BUG: fix a regression where `np.linspace`'s num argument would be ignored for `unyt_array`s
yt-project · Dec 25, 2024 · bf8e8ac · bf8e8ac
2 parents d78655f + c25ed28
commit bf8e8ac
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,