diff --git a/src/databricks/sql/parameters/native.py b/src/databricks/sql/parameters/native.py index b7c448254..f64eff548 100644 --- a/src/databricks/sql/parameters/native.py +++ b/src/databricks/sql/parameters/native.py @@ -63,9 +63,9 @@ class DatabricksSupportedType(Enum): bool, decimal.Decimal, None, - list, - dict, - tuple, + list["TAllowedParameterValue"], + "TParameterDict", + tuple["TAllowedParameterValue", ...], ]