diff --git a/ingestion/setup.py b/ingestion/setup.py index 1286dd8fde0..bf45e4ab2c5 100644 --- a/ingestion/setup.py +++ b/ingestion/setup.py @@ -38,7 +38,7 @@ base_requirements = { "email-validator>=1.0.3", "wheel~=0.36.2", "python-jose==3.3.0", - "sqlalchemy>=1.3.24", + "sqlalchemy>=1.4.0", "sql-metadata~=2.0.0", "requests~=2.26", "cryptography", @@ -63,6 +63,11 @@ base_plugins = { "sql-metadata~=2.0.0", } plugins: Dict[str, Set[str]] = { + "airflow-container": { + "marshmallow-sqlalchemy>=0.26.0", + "SQLAlchemy-Utils>=0.38.0", + "pymysql>=1.0.2", + }, "amundsen": {"neo4j~=4.4.0"}, "athena": {"PyAthena[SQLAlchemy]"}, "azuresql": {"pyodbc"},