Fixed versions (#70)

Author: RHRolun

1_experiment_train.ipynb

@@ -31,7 +31,7 @@
    },
    "outputs": [],
    "source": [
-    "!pip install onnx onnxruntime tf2onnx"
+    "!pip install onnx==1.17.0 onnxruntime==1.20.1 tf2onnx==1.16.1"
    ]
   },
   {

2_save_model.ipynb

@@ -24,7 +24,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install boto3 botocore"
+    "!pip install boto3==1.35.55 botocore==1.35.55"
    ]
   },
   {

7_get_data_train_upload.yaml

@@ -90,8 +90,8 @@ 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' 'onnxruntime'\
-          \ 'tf2onnx' && \"$0\" \"$@\"\n"
+          \  python3 -m pip install --quiet --no-warn-script-location 'onnx==1.17.0' 'onnxruntime==1.20.1'\
+          \ 'tf2onnx==1.16.1' && \"$0\" \"$@\"\n"
         - sh
         - -ec
         - 'program_path=$(mktemp -d)
@@ -170,7 +170,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 'boto3' 'botocore'\
+          \  python3 -m pip install --quiet --no-warn-script-location 'boto3==1.35.55' 'botocore==1.35.55'\
           \ && \"$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", "onnxruntime", "tf2onnx"],
+    packages_to_install=["onnx==1.17.0", "onnxruntime==1.20.1", "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
@@ -117,7 +117,7 @@ def train_model(train_data_input_path: InputPath(), validate_data_input_path: In
 
 @dsl.component(
     base_image="quay.io/modh/runtime-images:runtime-cuda-tensorflow-ubi9-python-3.9-2024a-20240523",
-    packages_to_install=["boto3", "botocore"]
+    packages_to_install=["boto3==1.35.55", "botocore==1.35.55"]
 )
 def upload_model(input_model_path: InputPath()):
     import os