Update tests/continuous/test_standard.py

Co-authored-by: Nicholas Loveday <[email protected]> Signed-off-by: Tennessee Leeuwenburg <[email protected]>
nci · Sep 19, 2024 · 234a671 · 234a671
1 parent 52df85c
commit 234a671
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/continuous/test_standard.py b/tests/continuous/test_standard.py
@@ -931,5 +931,5 @@ def test_kge_dask():
     result = scores.continuous.kge(fcst, obs)
     assert isinstance(result.data, dask.array.Array)  # type: ignore
     result = result.compute()  # type: ignore
-    assert isinstance(result.data, np.ndarray)
+    assert isinstance(result.data, (np.ndarray, np.generic))
     xr.testing.assert_equal(result, EXP_KGE_REDUCE_ALL)