fix(trainer): Updated requirements

2023-08-21 10:29:39 +02:00 · 2023-08-21 10:29:39 +02:00 · e2c7d63889
commit e2c7d63889
parent 1f1e4703c2
1 changed files with 1 additions and 0 deletions
--- 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