Merge branch 'gc_allow_dataarray' of https://github.com/mathause/mesmer…

… into gc_allow_dataarray
MESMER-group · Sep 21, 2023 · 76d9914 · 76d9914
2 parents 85d4346 + 179e1ea
commit 76d9914
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/mesmer/core/computation.py b/mesmer/core/computation.py
@@ -19,7 +19,7 @@ def gaspari_cohn(r):
 
     Returns
     -------
-    out : xr.DataArray
+    out : xr.DataArray, , np.array
         Gaspari-Cohn correlation function
 
     Notes
@@ -49,7 +49,6 @@ def gaspari_cohn(r):
         return _gaspari_cohn_np(r)
 
     out = _gaspari_cohn_np(r.values)
-
     out = xr.DataArray(out, dims=r.dims, coords=r.coords, attrs=r.attrs)
 
     return out

diff --git a/tests/unit/test_computation.py b/tests/unit/test_computation.py
@@ -27,11 +27,9 @@ def test_gaspari_cohn():
 
     expected = np.array([0.68489583, 1.0, 0.68489583, 0.20833333, 0.01649306, 0.0])
     expected = expected.reshape(2, 3)
-
     expected = xr.DataArray(expected, dims=dims, coords=coords, attrs=attrs)
 
     xr.testing.assert_allclose(expected, result, rtol=1e-6)
-
     assert result.attrs == attrs