Fix comments

databricks · Nov 1, 2023 · 53b55cf · 53b55cf
1 parent e16caf5
commit 53b55cf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/databricks/sdk/dbutils.py b/databricks/sdk/dbutils.py
@@ -174,7 +174,9 @@ def get(self, taskKey: str, key: str, default: any = None, debugValue: any = Non
             Returns `debugValue` if present, throws an error otherwise as this implementation is always run outside of a job run
             """
             if debugValue is None:
-                raise ValueError('No debug value set for task value, when outside of job run')
+                raise TypeError(
+                    'Must pass debugValue when calling get outside of a job context. debugValue cannot be None.'
+                )
             return debugValue
 
         def set(self, key: str, value: any) -> None:

diff --git a/tests/test_dbutils.py b/tests/test_dbutils.py
@@ -215,12 +215,14 @@ def test_jobs_task_values_get(dbutils):
 
     dbutils.jobs.taskValues.set('key', 'value')
 
+    # Expect `get` to always return the `debugValue`` when calling outside of a job context and not what was previously set using `set`
     assert dbutils.jobs.taskValues.get('taskKey', 'key', debugValue='debug') == 'debug'
 
 
 def test_jobs_task_values_get_throws(dbutils):
     try:
         dbutils.jobs.taskValues.get('taskKey', 'key')
         assert False
-    except ValueError as e:
-        assert str(e) == 'No debug value set for task value, when outside of job run'
+    except TypeError as e:
+        assert str(
+            e) == 'Must pass debugValue when calling get outside of a job context. debugValue cannot be None.'