diff --git a/providers/teradata/src/airflow/providers/teradata/operators/bteq.py b/providers/teradata/src/airflow/providers/teradata/operators/bteq.py index 2557a525a1b78..847ff1734c9c7 100644 --- a/providers/teradata/src/airflow/providers/teradata/operators/bteq.py +++ b/providers/teradata/src/airflow/providers/teradata/operators/bteq.py @@ -17,6 +17,7 @@ # under the License. from __future__ import annotations +from collections.abc import Sequence from typing import TYPE_CHECKING, Literal from airflow.providers.teradata.utils.bteq_util import ( @@ -75,7 +76,7 @@ class BteqOperator(BaseOperator): :param timeout_rc: Return code to use if the BTEQ execution fails due to a timeout. To allow DAG execution to continue after a timeout, include this value in `bteq_quit_rc`. If not specified, a timeout will raise an exception and stop the DAG. """ - template_fields = "sql" + template_fields: Sequence[str] = ("sql",) ui_color = "#ff976d" def __init__( diff --git a/providers/teradata/tests/unit/teradata/operators/test_bteq.py b/providers/teradata/tests/unit/teradata/operators/test_bteq.py index a099576960f7b..86bedcf8438a3 100644 --- a/providers/teradata/tests/unit/teradata/operators/test_bteq.py +++ b/providers/teradata/tests/unit/teradata/operators/test_bteq.py @@ -144,8 +144,7 @@ def test_on_kill_not_initialized(self): def test_template_fields(self): # Verify template fields are defined correctly - print(BteqOperator.template_fields) - assert BteqOperator.template_fields == "sql" + assert BteqOperator.template_fields == ("sql",) def test_execute_raises_if_no_sql_or_file(self): op = BteqOperator(task_id="fail_case", teradata_conn_id="td_conn")