Skip to content

Commit 420e70f

Browse files
committed
fix: skip test since histogram functionality not implemented on torch cuda
1 parent 89667cf commit 420e70f

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tests/wind/test_wind.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,15 @@ def test_coriolis(lat, v_ref, xp, device):
204204

205205

206206
# histogram2d is not available in torch, so we skip this test for now
207-
@pytest.mark.parametrize("xp, device", NAMESPACE_DEVICES)
207+
@pytest.mark.parametrize(
208+
"xp, device",
209+
list(
210+
filter(
211+
lambda x: not (x[0]._earthkit_array_namespace_name == "torch" and "cuda" in x[1]),
212+
NAMESPACE_DEVICES,
213+
)
214+
),
215+
)
208216
@pytest.mark.parametrize(
209217
"sp,d,sectors,sp_bins,percent,v_ref,dir_bin_ref",
210218
[

0 commit comments

Comments
 (0)