Merge remote-tracking branch 'upstream/main' into rhoai-3.0

Author: moulalis

jupyter/rocm/tensorflow/ubi9-python-3.12/pylock.toml

@@ -8,15 +8,17 @@ dependencies = [
 
     # ROCm TensorFlow packages
     # https://rocm.docs.amd.com/projects/install-on-linux/en/latest/install/3rd-party/tensorflow-install.html#using-a-wheels-package
-    "tensorflow-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4/tensorflow_rocm-2.18.1-cp312-cp312-manylinux_2_28_x86_64.whl#sha256=814a5e4842b0c92d63c7a0bb4df0baf51ff2db8615535d83fe8958204b840598",
-    "tensorboard~=2.18.0", # tensorflow-rocm==2.18.1 is available and tensorflow-rocm==2.18.1 depends on tensorboard>=2.18,<2.19
+    # TODO(jdanek): consider adding #sha256=814a5e4842b0c92d63c7a0bb4df0baf51ff2db8615535d83fe8958204b840598
+    "tensorflow-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.3/tensorflow_rocm-2.17.0-cp312-cp312-manylinux_2_28_x86_64.whl",
+    # tensorflow-rocm==2.x needs tensorboard>=2.x,<2.x+1
+    "tensorboard~=2.17.0",
 
     # Datascience and useful extensions
     "boto3~=1.40.52",
     "kafka-python-ng~=2.2.3",
     "kfp~=2.14.6",
     "matplotlib~=3.10.7",
-    "numpy~=2.0.2",
+    "numpy~=1.26.4",
     "pandas~=2.3.3",
     "plotly~=6.3.1",
     "scikit-learn~=1.7.2",
@@ -25,7 +27,6 @@ dependencies = [
     "onnxconverter-common~=1.13.0", # Required for skl2onnx, as upgraded version is not compatible with protobuf
     "codeflare-sdk~=0.32.0",
     "kubeflow-training==1.9.3",
-    "feast~=0.55.0",
 
     # DB connectors
     "pymongo~=4.15.3",
@@ -61,6 +62,11 @@ override-dependencies = [
     "protobuf==6.31.1"
 ]
 
+constraint-dependencies = [
+    # AttributeError: module 'ml_dtypes' has no attribute 'float4_e2m1fn'
+    "onnx<1.19.0",
+]
+
 environments = [
     "sys_platform == 'linux' and implementation_name == 'cpython'",
 ]

jupyter/rocm/tensorflow/ubi9-python-3.12/pyproject.toml

@@ -20,27 +20,26 @@ spec:
         # language=json
         opendatahub.io/notebook-software: |
           [
-            {"name": "ROCm", "version": "v6.4"},
+            {"name": "ROCm", "version": "v6.3"},
             {"name": "Python", "version": "v3.12"},
-            {"name": "TensorFlow-ROCm", "version": "2.18"}
+            {"name": "TensorFlow-ROCm", "version": "2.17"}
           ]
         # language=json
         opendatahub.io/notebook-python-dependencies: |
           [
             {"name": "JupyterLab","version": "4.4"},
-            {"name": "TensorFlow-ROCm", "version": "2.18"},
-            {"name": "Tensorboard", "version": "2.18"},
+            {"name": "TensorFlow-ROCm", "version": "2.17"},
+            {"name": "Tensorboard", "version": "2.17"},
             {"name": "Kafka-Python-ng", "version": "2.2"},
             {"name": "Matplotlib", "version": "3.10"},
-            {"name": "Numpy", "version": "2.0"},
+            {"name": "Numpy", "version": "1.26"},
             {"name": "Pandas", "version": "2.3"},
             {"name": "Scikit-learn", "version": "1.7"},
             {"name": "Scipy", "version": "1.16"},
             {"name": "Odh-Elyra", "version": "4.3"},
             {"name": "PyMongo", "version": "4.15"},
             {"name": "Pyodbc", "version": "5.2"},
             {"name": "Codeflare-SDK", "version": "0.32"},
-            {"name": "Feast", "version": "0.55"},
             {"name": "Sklearn-onnx", "version": "1.19"},
             {"name": "Psycopg", "version": "3.2"},
             {"name": "MySQL Connector/Python", "version": "9.4"}