diff --git a/‎tests/unit/test_session.py‎
Lines changed: 1 addition & 0 deletions b/‎tests/unit/test_session.py‎
Lines changed: 1 addition & 0 deletions
@@ -800,6 +800,7 @@ def test_enable_geospatial_support_matches_real_kernel_signature(self):
             _kernel_geospatial_kwargs,
             _kernel_session_accepts_kwarg,
         )
+        from databricks.sql.exc import NotSupportedError
 
         # The ordinary kernel unit-test tier installs the latest published
         # wheel, which may lag the KERNEL_REV source pin while the matching
Original file line number	Diff line number	Diff line change
`@@ -800,6 +800,7 @@ def test_enable_geospatial_support_matches_real_kernel_signature(self):`
`800`	`800`	`_kernel_geospatial_kwargs,`
`801`	`801`	`_kernel_session_accepts_kwarg,`
`802`	`802`	`)`
	`803`	`+ from databricks.sql.exc import NotSupportedError`
`803`	`804`
`804`	`805`	`# The ordinary kernel unit-test tier installs the latest published`
`805`	`806`	`# wheel, which may lag the KERNEL_REV source pin while the matching`