MAINT: cherry pick fix from 372dfbd

mj-will · Aug 8, 2024 · f57d392 · f57d392
1 parent ed688d1
commit f57d392
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/test_experimental/test_flowmodel/test_clustering.py b/tests/test_experimental/test_flowmodel/test_clustering.py
@@ -9,6 +9,7 @@ def cfm():
     return create_autospec(CFM)
 
 
+@pytest.mark.requires("faiss")
 def test_init(cfm, tmp_path, caplog):
     caplog.set_level("DEBUG")
     flow_config = {}
@@ -30,6 +31,7 @@ def test_init(cfm, tmp_path, caplog):
     assert "faiss version" in str(caplog.text)
 
 
+@pytest.mark.requires("faiss")
 @pytest.mark.integration_test
 def test_clustering_integration(tmp_path, caplog):
     caplog.set_level("DEBUG")

diff --git a/.../test_sampling/test_starndard_sampling.py → ...l/test_sampling/test_standard_sampling.py b/.../test_sampling/test_starndard_sampling.py → ...l/test_sampling/test_standard_sampling.py
@@ -4,6 +4,7 @@
 from nessai.experimental.proposal.clustering import ClusteringFlowProposal
 
 
+@pytest.mark.requires("faiss")
 @pytest.mark.slow_integration_test
 def test_sampling_with_clusteringflowproposal(integration_model, tmp_path):
     fp = FlowSampler(