Fix: incomplete ColumnMapping type hint

georgesittas · georgesittas · commit ad1ade1e5962 · 2026-04-08T19:21:55.000+03:00
diff --git a/sqlglot-integration-tests b/sqlglot-integration-tests
@@ -1 +1 @@
-Subproject commit 84deb90577bc67fec20675100e58ae113f231a7e
+Subproject commit 52937ac8780c0aba9a9ce7df1164c8171915a628
diff --git a/sqlglot/schema.py b/sqlglot/schema.py
@@ -18,7 +18,7 @@
     from collections.abc import Sequence
     from typing_extensions import Unpack
 
-    ColumnMapping = t.Union[dict[str, str], str, list[str]]
+    ColumnMapping = dict[str, object]
 
 
 @trait
@@ -708,13 +708,7 @@ def ensure_schema(
     return MappingSchema(schema, **kwargs)
 
 
-@t.overload
-def ensure_column_mapping(mapping: list[str] | None) -> dict[str, None]: ...
-@t.overload
-def ensure_column_mapping(mapping: dict[str, str]) -> dict[str, str]: ...
-@t.overload
-def ensure_column_mapping(mapping: ColumnMapping | None) -> dict[str, None] | dict[str, str]: ...
-def ensure_column_mapping(mapping: ColumnMapping | None) -> dict[str, None] | dict[str, str]:
+def ensure_column_mapping(mapping: ColumnMapping | None) -> ColumnMapping:
     if mapping is None:
         return {}
     elif isinstance(mapping, dict):
