Try to skip tests on GPU

Lightning-AI · Nov 13, 2022 · 55c499e · 55c499e
1 parent 4d130dd
commit 55c499e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/unittests/nominal/test_cramers.py b/tests/unittests/nominal/test_cramers.py
@@ -90,6 +90,9 @@ def _dython_cramers_v_matrix(matrix, bias_correction, nan_strategy, nan_replace_
 @pytest.mark.skipif(
     _compare_version("pandas", operator.lt, "1.3.2"), reason="`dython` package requires `pandas>=1.3.2`"
 )
+@pytest.mark.skipif(  # TODO: testing on CUDA fails with pandas 1.3.5, and newer is not available for python 3.7
+    torch.cuda.is_available(), reason="Tests fail on CUDA with the most up-to-data available pandas"
+)
 @pytest.mark.parametrize(
     "preds, target",
     [