From 7718ca4bf6042f2af86264038c7d7b220be1f6bb Mon Sep 17 00:00:00 2001 From: Tom White Date: Thu, 23 Nov 2023 12:35:58 +0000 Subject: [PATCH] Fix dtype in test for mean --- cubed/tests/test_gufunc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cubed/tests/test_gufunc.py b/cubed/tests/test_gufunc.py index f68e3b87..a06d2e65 100644 --- a/cubed/tests/test_gufunc.py +++ b/cubed/tests/test_gufunc.py @@ -22,7 +22,7 @@ def stats(x): r = np.random.normal(size=(10, 20, 30)) a = cubed.from_array(r, chunks=(5, 5, 30), spec=spec) actual = apply_gufunc(stats, "(i)->()", a, output_dtypes="f", vectorize=vectorize) - expected = np.mean(r, axis=-1) + expected = np.mean(r, axis=-1, dtype=np.float32) assert actual.compute().shape == expected.shape assert_allclose(actual.compute(), expected)