From c25ed28655dee51f8aa6fbb40bc7980e07fe9727 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9ment=20Robert?= Date: Wed, 25 Dec 2024 18:18:20 +0100 Subject: [PATCH] BUG: fix a regression where np.linspace's num argument would be ignored for unyt_arrays --- unyt/_array_functions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,