diff --git a/unyt/_array_functions.py b/unyt/_array_functions.py index 03a4875d..df145640 100644 --- 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,