Added labels for classical-ml (#84)

Co-authored-by: Tyler Titsworth <[email protected]>
intel · Jun 5, 2024 · 02a14d0 · 02a14d0
1 parent 02168d8
commit 02a14d0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/classical-ml/docker-compose.yaml b/classical-ml/docker-compose.yaml
@@ -33,6 +33,14 @@ services:
         REGISTRY: ${REGISTRY}
         REPO: ${REPO}
       context: .
+      labels:
+        dependency.python: ${PYTHON_VERSION:-3.10}
+        dependency.python.pip: requirements.txt
+        docs: classical-ml
+        org.opencontainers.base.name: "intel/python:3.10-core"
+        org.opencontainers.image.name: "intel/intel-optimized-ml"
+        org.opencontainers.image.title: "Intel® Optimized ML Base Image"
+        org.opencontainers.image.version: ${SKLEARN_VERSION:-2024.4.0}-${PACKAGE_OPTION:-pip}-base
       target: ml-base-${PACKAGE_OPTION:-pip}
     command: >
       bash -c "python -c 'import sklearnex, sklearn; import xgboost as xgb; print(\"Scikit version:\", sklearn.__version__, \"\\nXGBoost version:\", xgb.__version__)'"
@@ -42,6 +50,11 @@ services:
     pull_policy: always
   jupyter:
     build:
+      labels:
+        dependency.python.pip: jupyter-requirements.txt
+        org.opencontainers.base.name: "intel/intel-optimized-ml:${SKLEARN_VERSION:-2024.4.0}-${PACKAGE_OPTION:-pip}-base"
+        org.opencontainers.image.title: "Intel® Optimized ML Jupyter Base Image"
+        org.opencontainers.image.version: ${SKLEARN_VERSION:-2024.4.0}-${PACKAGE_OPTION:-pip}-jupyter
       target: jupyter
     command: >
       bash -c "python -m jupyter --version"

diff --git a/docs/scripts/hook.py b/docs/scripts/hook.py
@@ -33,7 +33,8 @@ def create_support_matrix():
     compose_to_csv("pytorch", None)
     compose_to_csv("pytorch", "serving")
     compose_to_csv("tensorflow", None)
-    # compose_to_csv("classical-ml")
+    compose_to_csv("classical-ml", None)
+
     # get_repo(models)
     # compose_to_csv("preset/data-analytics", "data_analytics")
     # compose_to_csv("preset/classical-ml", "classical_ml")