Fix mypy static type failures

Jesse Whitehouse · Jesse Whitehouse · commit 00d8a494f911 · 2022-12-29T23:28:22.000-06:00
Signed-off-by: Jesse Whitehouse &lt;jesse.whitehouse@databricks.com&gt;
diff --git a/src/databricks/sql/client.py b/src/databricks/sql/client.py
@@ -300,7 +300,7 @@ def _check_not_closed(self):
         if not self.open:
             raise Error("Attempting operation on closed cursor")
 
-    def _handle_staging_operation(self, staging_allowed_local_path: Union[List, str]):
+    def _handle_staging_operation(self, staging_allowed_local_path: Union[None, str, List[str]]):
         """Fetch the HTTP request instruction from a staging ingestion command
         and call the designated handler.
 
@@ -319,7 +319,9 @@ def _handle_staging_operation(self, staging_allowed_local_path: Union[List, str]
 
         abs_staging_allowed_local_paths = [os.path.abspath(i) for i in _staging_allowed_local_paths]
 
+        assert self.active_result_set is not None
         row = self.active_result_set.fetchone()
+        assert row is not None
 
         # Must set to None in cases where server response does not include localFile
         abs_localFile = None
diff --git a/src/databricks/sql/thrift_backend.py b/src/databricks/sql/thrift_backend.py
@@ -62,7 +62,7 @@ def __init__(
         http_path: str,
         http_headers,
         auth_provider: AuthProvider,
-        staging_allowed_local_path: Union[str, List] = None,
+        staging_allowed_local_path: Union[None, str, List[str]] = None,
         **kwargs,
     ):
         # Internal arguments in **kwargs:
