51 lines
1.6 KiB
Bash
Executable File
51 lines
1.6 KiB
Bash
Executable File
set -e
|
|
set -o xtrace
|
|
|
|
############################################################
|
|
# Help #
|
|
############################################################
|
|
Help() {
|
|
# Display Help
|
|
echo "ModelRegistry install script."
|
|
echo
|
|
echo "Syntax: [-n NAMESPACE] [-i IMAGE]"
|
|
echo "options:"
|
|
echo " n Namespace."
|
|
echo " i Model registry image."
|
|
echo
|
|
}
|
|
|
|
MR_ROOT="$(dirname "$0")/../.."
|
|
|
|
namespace=kubeflow
|
|
image=ghcr.io/kubeflow/model-registry/server:latest
|
|
while getopts ":hn:i:" option; do
|
|
case $option in
|
|
h) # display Help
|
|
Help
|
|
exit;;
|
|
n) # override namespace
|
|
namespace=$OPTARG;;
|
|
i) # override model registry image
|
|
image=$OPTARG;;
|
|
\?) # Invalid option
|
|
echo "Error: Invalid option"
|
|
exit;;
|
|
esac
|
|
done
|
|
|
|
# Create namespace if not already existing
|
|
if ! kubectl get namespace "$namespace" &> /dev/null; then
|
|
kubectl create namespace $namespace
|
|
fi
|
|
# Apply model-registry kustomize manifests
|
|
echo Using model registry image: $image
|
|
cd $MR_ROOT/manifests/kustomize/base && kustomize edit set image ghcr.io/kubeflow/model-registry/server:latest=${image} && \
|
|
kustomize edit set namespace $namespace && cd -
|
|
cd $MR_ROOT/manifests/kustomize/overlays/db && kustomize edit set namespace $namespace && cd -
|
|
kubectl -n $namespace apply -k "$MR_ROOT/manifests/kustomize/overlays/db"
|
|
|
|
# Wait for model registry deployment
|
|
modelregistry=$(kubectl get pod -n $namespace --selector="component=model-registry-server" --output jsonpath='{.items[0].metadata.name}')
|
|
kubectl wait --for=condition=Ready pod/$modelregistry -n $namespace --timeout=6m
|