From 00552f6d3641ac446ec64e395a3364c810648e87 Mon Sep 17 00:00:00 2001
From: Pere Miquel Brull <peremiquelbrull@gmail.com>
Date: Mon, 28 Feb 2022 17:11:18 +0100
Subject: [PATCH] Fix #3012 - Dependency breaking Airflow Container (#3015)

Fix #3012 - Dependency breaking Airflow Container
---
 ingestion/setup.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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"},