fix: update onnxruntime version (#82)

Signed-off-by: kramaranya <[email protected]>

Author: kramaranya

1_experiment_train.ipynb

@@ -30,9 +30,7 @@
     "scrolled": true
    },
    "outputs": [],
-   "source": [
-    "!pip install onnx==1.17.0 onnxruntime==1.20.1 tf2onnx==1.16.1"
-   ]
+   "source": "!pip install onnx==1.17.0 onnxruntime==1.19.2 tf2onnx==1.16.1"
   },
   {
    "cell_type": "markdown",

7_get_data_train_upload.yaml

@@ -90,7 +90,7 @@ deploymentSpec:
           \ python3 -m ensurepip --user || apt-get install python3-pip\nfi\n\nPIP_DISABLE_PIP_VERSION_CHECK=1\
           \ python3 -m pip install --quiet --no-warn-script-location 'kfp==2.8.0'\
           \ '--no-deps' 'typing-extensions>=3.7.4,<5; python_version<\"3.9\"'  &&\
-          \  python3 -m pip install --quiet --no-warn-script-location 'onnx==1.17.0' 'onnxruntime==1.20.1'\
+          \  python3 -m pip install --quiet --no-warn-script-location 'onnx==1.17.0' 'onnxruntime==1.19.2'\
           \ 'tf2onnx==1.16.1' && \"$0\" \"$@\"\n"
         - sh
         - -ec

pipeline/7_get_data_train_upload.py

@@ -23,7 +23,7 @@ def get_data(train_data_output_path: OutputPath(), validate_data_output_path: Ou
 
 @dsl.component(
     base_image="quay.io/modh/runtime-images:runtime-cuda-tensorflow-ubi9-python-3.9-2024a-20240523",
-    packages_to_install=["onnx==1.17.0", "onnxruntime==1.20.1", "tf2onnx==1.16.1"],
+    packages_to_install=["onnx==1.17.0", "onnxruntime==1.19.2", "tf2onnx==1.16.1"],
 )
 def train_model(train_data_input_path: InputPath(), validate_data_input_path: InputPath(), model_output_path: OutputPath()):
     import numpy as np