From fcffe8e1e9cefe499671fa6ca8ae3f89c445c9c4 Mon Sep 17 00:00:00 2001
From: Mesh <meshach.ogunmodede@datakind.org>
Date: Thu, 5 Jun 2025 10:40:19 -0500
Subject: [PATCH] feat: added option for api auth

---
 src/webapp/databricks.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/webapp/databricks.py b/src/webapp/databricks.py
index 0b57421f..dc22e933 100644
--- a/src/webapp/databricks.py
+++ b/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