skip entire module

holoviz · Oct 22, 2024 · 61e55a1 · 61e55a1
1 parent 2819471
commit 61e55a1
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/lumen/tests/ai/test_utils.py b/lumen/tests/ai/test_utils.py
@@ -8,10 +8,13 @@
 
 from panel.chat import ChatStep
 
-from lumen.ai.utils import (
-    UNRECOVERABLE_ERRORS, clean_sql, describe_data, format_schema, get_schema,
-    render_template, report_error, retry_llm_output,
-)
+try:
+    from lumen.ai.utils import (
+        UNRECOVERABLE_ERRORS, clean_sql, describe_data, format_schema,
+        get_schema, render_template, report_error, retry_llm_output,
+    )
+except ImportError:
+    pytest.skip("Skipping tests that require lumen.ai", allow_module_level=True)
 
 
 def test_render_template_with_valid_template():