Tuple validation in OBJECT class

snowflakedb · Dec 16, 2024 · e1ca5aa · e1ca5aa
1 parent a171431
commit e1ca5aa
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/snowflake/sqlalchemy/custom_types.py b/src/snowflake/sqlalchemy/custom_types.py
@@ -63,6 +63,10 @@ class OBJECT(StructuredType):
     __visit_name__ = "OBJECT"
 
     def __init__(self, **items_types: Union[TypeEngine, Tuple[TypeEngine, bool]]):
+        for key, value in items_types.items():
+            if not isinstance(value, tuple):
+                items_types[key] = (value, False)
+
         self.items_types = items_types
         self.is_semi_structured = len(items_types) == 0
         super().__init__()