diff options
Diffstat (limited to 'sqlglot/schema.py')
-rw-r--r-- | sqlglot/schema.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/sqlglot/schema.py b/sqlglot/schema.py index f5d9f2b..8e39c7f 100644 --- a/sqlglot/schema.py +++ b/sqlglot/schema.py @@ -306,11 +306,11 @@ class MappingSchema(AbstractMappingSchema[t.Dict[str, str]], Schema): return self._type_mapping_cache[schema_type] -def ensure_schema(schema: t.Any) -> Schema: +def ensure_schema(schema: t.Any, dialect: DialectType = None) -> Schema: if isinstance(schema, Schema): return schema - return MappingSchema(schema) + return MappingSchema(schema, dialect=dialect) def ensure_column_mapping(mapping: t.Optional[ColumnMapping]): |