diff --git a/src/databricks/sql/parameters/native.py b/src/databricks/sql/parameters/native.py index b7c448254..762c00005 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"], + dict["TAllowedParameterValue", "TAllowedParameterValue"], + tuple["TAllowedParameterValue", ...], ]