From 47ac5f8ea2138f48c7d1a6fa75ed871dce5fad22 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9ment=20Robert?= Date: Sun, 15 Jan 2023 12:52:16 +0100 Subject: [PATCH] TST: fixup a doctest --- unyt/dask_array.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/unyt/dask_array.py b/unyt/dask_array.py index ef51960d..6d4cc0e3 100644 --- a/unyt/dask_array.py +++ b/unyt/dask_array.py @@ -650,10 +650,11 @@ def reduce_with_units(dask_func, unyt_dask_in, *args, **kwargs): Examples -------- - >>> from unyt import dask_array - >>> a = dask_array.dask.array.ones((10000,), chunks=(100,)) - >>> a = dask_array.unyt_from_dask(a, 'm') - >>> b = dask_array.reduce_with_units(dask_array.dask.array.median, a, axis=0) + >>> import dask.array + >>> from unyt.dask_array import unyt_from_dask, reduce_with_units + >>> a = dask.array.ones((10000,), chunks=(100,)) + >>> a = unyt_from_dask(a, 'm') + >>> b = reduce_with_units(dask.array.median, a, axis=0) >>> b.compute() unyt_quantity(1., 'm')