added logging

Author: mwojtyczka

src/databricks/labs/pytester/fixtures/connect.py

@@ -26,7 +26,6 @@ def test_databricks_connect(spark):
     """
     cluster_id = ws.config.cluster_id
     serverless_cluster_id = ws.config.serverless_compute_id
-    print(f"cluster id is: {serverless_cluster_id}")
 
     if not serverless_cluster_id:
         ensure_cluster_is_running(cluster_id, ws)
@@ -41,8 +40,10 @@ def test_databricks_connect(spark):
         )
 
         if serverless_cluster_id:
-            logging.debug("Using serverless compute")
+            logging.debug(f"Using serverless cluster id '{serverless_cluster_id}'")
             return DatabricksSession.builder.serverless(True).getOrCreate()
+
+        logging.debug(f"Using cluster id '{cluster_id}'")
         return DatabricksSession.builder.sdkConfig(ws.config).getOrCreate()
     except ImportError:
         skip("Please run `pip install databricks-connect`")