requirments updated to get

eflows4hpc · Dec 15, 2023 · 5522ea3 · 5522ea3
1 parent 5c9f1b7
commit 5522ea3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/mlflow_stagein.py b/mlflow_stagein.py
@@ -22,7 +22,7 @@
         "host": Param(default="", type="string"),
         "port": Param(type="integer", default=22),
         "login": Param(default="", type="string"),
-        "mlflow_runid": Param(type="string", description="run from which model should be staged-in"),
+        "mlflow_runid": Param(default="", type="string", description="run from which model should be staged-in"),
         "mlflow_modelpath": Param(type="string", default="model/model.pkl")
     },
 )

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-apache-airflow==2.6.3
+apache-airflow==2.7.0
 apache-airflow-providers-ssh
 apache-airflow-providers-hashicorp
 boto3==1.24.32
@@ -8,4 +8,5 @@ webdavclient3
 airflow-datacat-integration>=0.1.4
 virtualenv
 pyunicore==0.15.0
-python-gitlab==3.15.0
+python-gitlab==3.15.0
+mlflow==1.30.0