From e2c7d638895ee92242dbbfd6a3f4b6b8afbd5174 Mon Sep 17 00:00:00 2001
From: MauricioGarciaS <47052044+MauricioGarciaS@users.noreply.github.com>
Date: Mon, 21 Aug 2023 10:29:39 +0200
Subject: [PATCH] fix(trainer): Updated requirements

---
 ee/recommendation/ml_trainer/entrypoint.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ee/recommendation/ml_trainer/entrypoint.sh b/ee/recommendation/ml_trainer/entrypoint.sh
index af93b034b..b917e44c6 100755
--- a/ee/recommendation/ml_trainer/entrypoint.sh
+++ b/ee/recommendation/ml_trainer/entrypoint.sh
@@ -6,6 +6,7 @@ find airflow/ -type f -name "*.cfg" -exec sed -i "s/{{pg_port_airflow}}/${pg_por
 find airflow/ -type f -name "*.cfg" -exec sed -i "s/{{pg_dbname_airflow}}/${pg_dbname_airflow}/g" {} \;
 find airflow/ -type f -name "*.cfg" -exec sed -i "s#{{airflow_secret_key}}#${airflow_secret_key}#g" {} \;
 export MLFLOW_TRACKING_URI=postgresql+psycopg2://${pg_user_ml}:${pg_password_ml}@${pg_host_ml}:${pg_port_ml}/${pg_dbname_ml}
+mlflow db upgrade
 git init airflow/dags
 # Airflow setup
 # airflow db init