Merge pull request #103 from datakind/develop

feat: added option for api auth

Author: Mesh-ach

src/webapp/databricks.py

@@ -8,6 +8,7 @@
     Format,
     ExecuteStatementRequestOnWaitTimeout,
     Disposition,
+    StatementState,
 )
 from .config import databricks_vars, gcs_vars
 from .utilities import databricksify_inst_name, SchemaType
@@ -241,7 +242,7 @@ def fetch_table_data(
             raise ValueError(f"Databricks API call failed: {e}")
 
         # Check if the query execution was successful
-        if response.status.state != "SUCCEEDED":
+        if response.status.state != StatementState.SUCCEEDED:
             error_message = (
                 response.status.error.message
                 if response.status.error