Hello @Kayla , I hope this message finds you well.

Have you tried the STATEMENT_TIMEOUT?
The STATEMENT_TIMEOUT configuration parameter sets a timeout value in seconds. Any Databricks SQL statement with a wall clock run time exceeding the value is timed out.
You can set this parameter at the session level using the SET statement and at the global level using Configure SQL parameters or SQL Warehouses API.

doc.: https://docs.databricks.com/en/sql/language-manual/parameters/statement_timeout.html#statement_timeo...

Best regards,
Lucas Rocha