Don't initialise spark progress widget if it's not supported (#1402)

## Changes Don't initialise spark progress widget if it's not supported ## Tests Manually
databricks · Oct 23, 2024 · afe75eb · afe75eb
1 parent 261e06c
commit afe75eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/packages/databricks-vscode/package.json b/packages/databricks-vscode/package.json
@@ -882,7 +882,7 @@
                     "databricks.connect.progress": {
                         "type": "boolean",
                         "default": true,
-                        "description": "Show PySpark progress bar when using Databricks Connect."
+                        "markdownDescription": "Show PySpark progress bar when using Databricks Connect. Requires `ipywidgets` package and `pyspark@3.5.0+`."
                     },
                     "databricks.ipythonDir": {
                         "type": "string",

diff --git a/packages/databricks-vscode/resources/python/00-databricks-init.py b/packages/databricks-vscode/resources/python/00-databricks-init.py
@@ -376,6 +376,9 @@ def register_spark_progress(spark, show_progress: bool):
         import ipywidgets as widgets
     except Exception as e:
         return
+
+    if not hasattr(spark, "clearProgressHandlers") or not hasattr(spark, "registerProgressHandler"):
+        return
 
     class Progress:
         SI_BYTE_SIZES = (1 << 60, 1 << 50, 1 << 40, 1 << 30, 1 << 20, 1 << 10, 1)