mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 03:55:04 -05:00
e522f22ce8
- poetry lock required to update package name - related Makefile and licenses update to keep consistency (will be removed once the PyPi package is available) - modified compile_torch_model to only accept datasets with one input mypy error with better numpy typing hints appeared closes #1252
103 lines
2.8 KiB
Bash
Executable File
103 lines
2.8 KiB
Bash
Executable File
#!/bin/bash -e
|
|
|
|
set -e
|
|
|
|
BASE_IMG_ENDPOINT_URL=
|
|
ENV_IMG_ENDPOINT_URL=
|
|
TOKEN=
|
|
ENV_DOCKERFILE=./docker/Dockerfile.concrete-framework-env
|
|
GITHUB_ENV_FILE=debug.txt
|
|
|
|
while [ -n "$1" ]
|
|
do
|
|
case "$1" in
|
|
"--base_img_url" )
|
|
shift
|
|
BASE_IMG_ENDPOINT_URL="$1"
|
|
;;
|
|
|
|
"--env_img_url" )
|
|
shift
|
|
ENV_IMG_ENDPOINT_URL="$1"
|
|
;;
|
|
|
|
"--token" )
|
|
shift
|
|
TOKEN="$1"
|
|
;;
|
|
|
|
"--github-env")
|
|
shift
|
|
GITHUB_ENV_FILE="$1"
|
|
;;
|
|
|
|
*)
|
|
echo "Unknown param : $1"
|
|
exit 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
BASE_JSON=$(curl \
|
|
-X GET \
|
|
-H "Accept: application/vnd.github.v3+json" \
|
|
-H "Authorization: token ${TOKEN}" \
|
|
"${BASE_IMG_ENDPOINT_URL}")
|
|
|
|
LATEST_BASE_IMG_JSON=$(echo "${BASE_JSON}" | jq -rc 'sort_by(.updated_at)[-1]')
|
|
|
|
echo "Latest base image json: ${LATEST_BASE_IMG_JSON}"
|
|
|
|
LATEST_BASE_IMG_TIMESTAMP=$(echo "${LATEST_BASE_IMG_JSON}" | jq -r '.updated_at')
|
|
|
|
ENV_JSON=$(curl \
|
|
-X GET \
|
|
-H "Accept: application/vnd.github.v3+json" \
|
|
-H "Authorization: token ${TOKEN}" \
|
|
"${ENV_IMG_ENDPOINT_URL}")
|
|
|
|
LATEST_ENV_IMG_JSON=$(echo "${ENV_JSON}" | \
|
|
jq -rc '.[] | select(.metadata.container.tags[] | contains("latest"))')
|
|
|
|
ENV_IMG_TAG=$(echo "${LATEST_ENV_IMG_JSON}" | \
|
|
jq -rc '.metadata.container.tags - ["latest"] | .[0]')
|
|
|
|
echo "env image tag: ${ENV_IMG_TAG}"
|
|
|
|
IFS='-' read -ra ENV_IMG_SPLIT <<< "${ENV_IMG_TAG}"
|
|
ENV_IMG_BASE_IMG_TAG="${ENV_IMG_SPLIT[0]}"
|
|
|
|
echo "env image base image tag: ${ENV_IMG_BASE_IMG_TAG}"
|
|
|
|
CURRENT_BASE_IMG_JSON=$(echo "${BASE_JSON}" | \
|
|
jq -rc ".[] | select(.metadata.container.tags[] | contains(\"${ENV_IMG_BASE_IMG_TAG}\"))")
|
|
|
|
echo "current base image json: ${CURRENT_BASE_IMG_JSON}"
|
|
|
|
CURRENT_BASE_IMG_TIMESTAMP=$(echo "${CURRENT_BASE_IMG_JSON}" | jq -r '.updated_at')
|
|
|
|
echo "Latest base timestamp: ${LATEST_BASE_IMG_TIMESTAMP}"
|
|
echo "Current base timestamp: ${CURRENT_BASE_IMG_TIMESTAMP}"
|
|
|
|
LATEST_BASE_IMG_TIMESTAMP=$(date -d "${LATEST_BASE_IMG_TIMESTAMP}" +%s)
|
|
CURRENT_BASE_IMG_DATE=$(date -d "${CURRENT_BASE_IMG_TIMESTAMP}" +%s)
|
|
|
|
echo "Base epoch: ${LATEST_BASE_IMG_TIMESTAMP}"
|
|
echo "Env epoch: ${CURRENT_BASE_IMG_DATE}"
|
|
|
|
if [[ "${LATEST_BASE_IMG_TIMESTAMP}" -gt "${CURRENT_BASE_IMG_DATE}" ]]; then
|
|
echo "Env image out of date, sending rebuild request."
|
|
NEW_BASE_IMG_TAG=$(echo "${LATEST_BASE_IMG_JSON}" | \
|
|
jq -rc '.metadata.container.tags - ["latest"] | .[0]')
|
|
echo "NEW_BASE_IMG_TAG=${NEW_BASE_IMG_TAG}" >> "${GITHUB_ENV_FILE}"
|
|
echo "New base img tag: ${NEW_BASE_IMG_TAG}"
|
|
TMP_DOCKER_FILE="$(mktemp)"
|
|
sed "s/\(FROM\ ghcr\.io\/zama-ai\/zamalang-compiler:\)\(.*\)/\1${NEW_BASE_IMG_TAG}/g" \
|
|
"${ENV_DOCKERFILE}" > "${TMP_DOCKER_FILE}"
|
|
cp -f "${TMP_DOCKER_FILE}" "${ENV_DOCKERFILE}"
|
|
rm -f "${TMP_DOCKER_FILE}"
|
|
else
|
|
echo "Image up to date, nothing to do."
|
|
fi
|