From 970ebbaae49a134bd72607b71a4936fbb8edab29 Mon Sep 17 00:00:00 2001 From: Timothy Tamm Date: Thu, 15 Jan 2026 13:04:03 -0800 Subject: [PATCH] Make conftest more robust --- .../default/template/{{.project_name}}/tests/conftest.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/libs/template/templates/default/template/{{.project_name}}/tests/conftest.py b/libs/template/templates/default/template/{{.project_name}}/tests/conftest.py index 72ebfeb566..23502fb28e 100644 --- a/libs/template/templates/default/template/{{.project_name}}/tests/conftest.py +++ b/libs/template/templates/default/template/{{.project_name}}/tests/conftest.py @@ -59,7 +59,8 @@ def _loader(filename: str): def _enable_fallback_compute(): """Enable serverless compute if no compute is specified.""" conf = WorkspaceClient().config - if conf.serverless_compute_id or conf.cluster_id or os.environ.get("SPARK_REMOTE"): + has_serverles_compute_id = hasattr(conf, "serverless_compute_id") and conf.serverless_compute_id + if has_serverles_compute_id or conf.cluster_id or os.environ.get("SPARK_REMOTE"): return url = "https://docs.databricks.com/dev-tools/databricks-connect/cluster-config" @@ -83,6 +84,8 @@ def _allow_stderr_output(config: pytest.Config): def pytest_configure(config: pytest.Config): """Configure pytest session.""" with _allow_stderr_output(config): + src_path = pathlib.Path(__file__).parent.parent / "src" + sys.path.insert(0, str(src_path)) _enable_fallback_compute() # Initialize Spark session eagerly, so it is available even when