diff --git a/meta.yaml b/meta.yaml index 2adcd35..52e4099 100644 --- a/meta.yaml +++ b/meta.yaml @@ -41,6 +41,7 @@ test: - cdo - esmf - hdf5 + - nco files: - setup.cfg - test diff --git a/src/climtas/event.py b/src/climtas/event.py index 14478ec..32587fd 100644 --- a/src/climtas/event.py +++ b/src/climtas/event.py @@ -183,6 +183,7 @@ def atleastn(da: xarray.DataArray, n: int, dim: str = "time") -> xarray.DataArra """ def atleastn_helper(array, axis, n): + axis = axis[0] count = numpy.zeros_like(numpy.take(array, 0, axis=axis), dtype="i4") mask = numpy.empty_like(numpy.take(array, 0, axis=axis), dtype="bool") mask = True diff --git a/test/test_regrid.py b/test/test_regrid.py index bc7d27d..a3886b9 100644 --- a/test/test_regrid.py +++ b/test/test_regrid.py @@ -294,10 +294,7 @@ def test_esmf_generate_weights(): @pytest.mark.parametrize( "weight_gen,weight_args", - [ - (esmf_generate_weights, {}), - (esmf_generate_weights, {"method": "patch"}), - ], + [(esmf_generate_weights, {}), (esmf_generate_weights, {"method": "patch"})], ) def test_nco(tmpdir, weight_gen, weight_args): alats = 10