Lint fix: types

Signed-off-by: Jesse Whitehouse <[email protected]>
databricks · Sep 29, 2023 · df582c7 · df582c7
1 parent 4a7588a
commit df582c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/databricks/sql/utils.py b/src/databricks/sql/utils.py
@@ -551,13 +551,13 @@ def infer_types(params: list[DbSqlParameter]):
     for param in params:
         _name: str = param.name
         _value: Any = param.value
-        _type: DbSqlType
+        _type: Union[DbSqlType, DbsqlDynamicDecimalType, Enum, None]
 
         if param.type:
             _type = param.type
         else:
             # figure out what type to use
-            _type = type_lookup_table.get(type(_value), False)
+            _type = type_lookup_table.get(type(_value), None)
             if not _type:
                 raise ValueError(
                     f"Could not infer parameter type from {type(param.value)} - {param.value}"