diff --git a/atlas_provider_sqlalchemy/ddl.py b/atlas_provider_sqlalchemy/ddl.py index c70d7d4..b31dde9 100644 --- a/atlas_provider_sqlalchemy/ddl.py +++ b/atlas_provider_sqlalchemy/ddl.py @@ -2,7 +2,7 @@ import importlib.util import inspect from pathlib import Path -from typing import Any, Protocol +from typing import Any, Protocol, Tuple, List import sqlalchemy as sa @@ -21,7 +21,7 @@ class ModelsNotFoundError(Exception): pass -def sqlalchemy_version() -> tuple[int, ...]: +def sqlalchemy_version() -> Tuple[int, ...]: """Get major and minor version of sqlalchemy.""" return tuple(int(x) for x in sa.__version__.split(".")) @@ -86,7 +86,7 @@ def dump(sql, *multiparams, **params): return metadata -def print_ddl(dialect_driver: str, models: list[DBTableDesc]) -> None: +def print_ddl(dialect_driver: str, models: List[DBTableDesc]) -> None: """Dump DDL statements for the metadata from the given models/tables to stdout.""" dump_ddl(dialect_driver=dialect_driver, metadata=models[0].metadata)