mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
Antonin StefanuttiandGitHub
7f057acd73
* chore(test): Support e2e cluster setup with Podman Signed-off-by: Antonin Stefanutti <[email protected]> * Fix KinD cluster creation for GPU E2E tests Signed-off-by: Antonin Stefanutti <[email protected]> --------- Signed-off-by: Antonin Stefanutti <[email protected]>
34 lines
1.1 KiB
Bash
34 lines
1.1 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# Copyright 2025 The Kubeflow Authors.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
# Function to load container image into KinD cluster
|
|
load_image_to_kind() {
|
|
local image_name="$1"
|
|
local cluster_name="${2:-}"
|
|
local cluster_arg=""
|
|
|
|
if [[ -n "${cluster_name}" ]]; then
|
|
cluster_arg="--name ${cluster_name}"
|
|
fi
|
|
|
|
echo "Loading image ${image_name} into KinD cluster${cluster_name:+ ${cluster_name}}"
|
|
if [[ "${CONTAINER_RUNTIME}" == "docker" ]]; then
|
|
${KIND} load docker-image "${image_name}" ${cluster_arg}
|
|
else
|
|
${CONTAINER_RUNTIME} save "${image_name}" -o /dev/stdout | ${KIND} load image-archive /dev/stdin ${cluster_arg}
|
|
fi
|
|
}
|