From 7764aaaadeb5e3852d49f6c9f17628365e109cf9 Mon Sep 17 00:00:00 2001 From: Andrey Velichkevich Date: Tue, 9 Mar 2021 01:33:23 +0000 Subject: [PATCH] Add Katib images to private ECR (#923) --- .../config/static_config/ECR_Resources.py | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/aws/IaC/CDK/test-infra/config/static_config/ECR_Resources.py b/aws/IaC/CDK/test-infra/config/static_config/ECR_Resources.py index 1822e5838fb..523a60afafd 100644 --- a/aws/IaC/CDK/test-infra/config/static_config/ECR_Resources.py +++ b/aws/IaC/CDK/test-infra/config/static_config/ECR_Resources.py @@ -11,4 +11,30 @@ # "kfserving/agent": "cdk-poc/kfserving/agent", # "pytorch-operator": "cdk-poc/pytorch-operator", # "tf-operator": "cdk-poc/tf-operator", + + # Katib images. + # Katib main components. + "katib-controller": "/katib/v1beta1/katib-controller", + "katib-db-manager": "/katib/v1beta1/katib-db-manager", + "katib-ui": "/katib/v1beta1/katib-ui", + "katib-cert-generator": "/katib/v1beta1/cert-generator", + # Katib Metric Collector list. + "katib-file-metrics-collector": "/katib/v1beta1/file-metrics-collector", + "katib-tfevent-metrics-collector": "/katib/v1beta1/tfevent-metrics-collector", + # Katib Suggestion list. + "katib-suggestion-hyperopt": "/katib/v1beta1/suggestion-hyperopt", + "katib-suggestion-chocolate": "/katib/v1beta1/suggestion-chocolate", + "katib-suggestion-skopt": "/katib/v1beta1/suggestion-skopt", + "katib-suggestion-hyperband": "/katib/v1beta1/suggestion-hyperband", + "katib-suggestion-goptuna": "/katib/v1beta1/suggestion-goptuna", + "katib-suggestion-enas": "/katib/v1beta1/suggestion-enas", + "katib-suggestion-darts": "/katib/v1beta1/suggestion-darts", + # Katib Early Stopping list. + "katib-earlystopping-medianstop": "/katib/v1beta1/earlystopping-medianstop", + # Katib Trial list. + "katib-trial-mxnet-mnist": "/katib/v1beta1/trial-mxnet-mnist", + "katib-trial-pytorch-mnist": "/katib/v1beta1/trial-pytorch-mnist", + "katib-trial-enas-cnn-cifar10-gpu": "/katib/v1beta1/trial-enas-cnn-cifar10-gpu", + "katib-trial-enas-cnn-cifar10-cpu": "/katib/v1beta1/trial-enas-cnn-cifar10-cpu", + "katib-trial-darts-cnn-cifar10": "/katib/v1beta1/trial-darts-cnn-cifar10", }