Fix dtype in test for mean

cubed-dev · Nov 23, 2023 · 7718ca4 · 7718ca4
1 parent 05c3dfe
commit 7718ca4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)