Added conditional skipifs for validation tests requiring GPUs

CosmoStat · Mar 5, 2024 · 6135a08 · 6135a08
1 parent 478aed1
commit 6135a08
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/wf_psf/tests/metrics_test.py b/src/wf_psf/tests/metrics_test.py
@@ -108,7 +108,7 @@ def main_metrics(training_params):
     return np.load(os.path.join(main_dir, metrics_filename), allow_pickle=True)[()]
 
 
-@pytest.mark.skip(reason="Requires gpu")
+@pytest.mark.skipif("GITHUB_ENV" in os.environ, reason="Skipping GPU tests in CI")
 def test_eval_metrics_polychromatic_lowres(
     training_params,
     weights_path_basename,
@@ -156,7 +156,7 @@ def test_eval_metrics_polychromatic_lowres(
     assert ratio_rel_std_rmse < tol
 
 
-@pytest.mark.skip(reason="Requires gpu")
+@pytest.mark.skipif("GITHUB_ENV" in os.environ, reason="Skipping GPU tests in CI")
 def test_evaluate_metrics_opd(
     training_params,
     weights_path_basename,
@@ -206,7 +206,7 @@ def test_evaluate_metrics_opd(
     assert ratio_rel_rmse_std_opd < tol
 
 
-@pytest.mark.skip(reason="Requires gpu")
+@pytest.mark.skipif("GITHUB_ENV" in os.environ, reason="Skipping GPU tests in CI")
 def test_eval_metrics_mono_rmse(
     training_params,
     weights_path_basename,
@@ -271,7 +271,7 @@ def test_eval_metrics_mono_rmse(
     assert ratio_rel_rmse_std_mono < tol
 
 
-@pytest.mark.skip(reason="Requires gpu")
+@pytest.mark.skipif("GITHUB_ENV" in os.environ, reason="Skipping GPU tests in CI")
 def test_evaluate_metrics_shape(
     training_params,
     weights_path_basename,

diff --git a/src/wf_psf/tests/train_test.py b/src/wf_psf/tests/train_test.py
@@ -66,7 +66,7 @@ def psf_model_dir():
     )
 
 
-@pytest.mark.skip(reason="Requires gpu")
+@pytest.mark.skipif("GITHUB_ENV" in os.environ, reason="Skipping GPU tests in CI")
 def test_train(
     training_params,
     training_data,