skipping fancy prmpting test module

ur-whitelab · Jun 25, 2024 · e807565 · e807565
1 parent 7f92e73
commit e807565
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_utils/test_query_filter.py b/tests/test_utils/test_query_filter.py
@@ -1,7 +1,15 @@
 # test_query_filter.py
 import pytest
 
-from mdagent.agent.query_filter import Example, FilteredQuery, Parameters, query_filter
+try:
+    from mdagent.agent.query_filter import (
+        Example,
+        FilteredQuery,
+        Parameters,
+        query_filter,
+    )
+except Exception:
+    pytest.skip(allow_module_level=True)
 
 
 @pytest.mark.skip(reason="We skip because query filter was moved to WIP branch.")