From 234a671ed2110e240c7f05dd9f4b688ae371d911 Mon Sep 17 00:00:00 2001 From: Tennessee Leeuwenburg <134973832+tennlee@users.noreply.github.com> Date: Thu, 19 Sep 2024 19:33:56 +1000 Subject: [PATCH] Update tests/continuous/test_standard.py Co-authored-by: Nicholas Loveday <48701367+nicholasloveday@users.noreply.github.com> Signed-off-by: Tennessee Leeuwenburg <134973832+tennlee@users.noreply.github.com> --- tests/continuous/test_standard.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/continuous/test_standard.py b/tests/continuous/test_standard.py index a6e5efd7..a98562d6 100644 --- 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)