From 02a14d02d59ab169c28b2c2658a2387e4b510faf Mon Sep 17 00:00:00 2001 From: Sharvil Shah Date: Tue, 4 Jun 2024 22:13:17 -0700 Subject: [PATCH] Added labels for classical-ml (#84) Co-authored-by: Tyler Titsworth --- classical-ml/docker-compose.yaml | 13 +++++++++++++ docs/scripts/hook.py | 3 ++- 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/classical-ml/docker-compose.yaml b/classical-ml/docker-compose.yaml index d6df1204..f7f1b1bc 100644 --- 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 index 0d2e0694..bbbf83cd 100644 --- 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")