CI/dependencies/download_models.sh

0001 #!/bin/bash
0002
0003 set -e
0004 set -u
0005 set -o pipefail
0006
0007 # Assert MODEL_STORAGE environment variable is set
0008 if [[ -z "${MODEL_STORAGE+x}" ]]; then
0009   echo "Error: MODEL_STORAGE environment variable is not set"
0010   exit 1
0011 fi
0012
0013 function download {
0014   URL=$1
0015   HASH=$2
0016
0017   tarname=$(basename "$URL")
0018   curl -SL "$URL" -o "$tarname"
0019   echo "$HASH $tarname" | sha256sum -c
0020   tar -xf "$tarname"
0021 }
0022
0023 mkdir -p "${MODEL_STORAGE}"
0024 cd "${MODEL_STORAGE}"
0025
0026 download \
0027   https://acts.web.cern.ch/ci/gnn/onnx_models_v01.tar \
0028   335d829439d9a5ae99ffc8f0bbf6a62119a140475e779e2ed00de21fdebb3cb4
0029
0030 download \
0031   https://acts.web.cern.ch/ci/gnn/torchscript_models_v01.tar \
0032   1185060ce697bbc96c9dc32b85e5f0eb4db1f64a645c0fc4d2cb2731cb2ef3dc
0033
0034 download \
0035   https://acts.web.cern.ch/ci/gnn/odd_module_map_v01.tar \
0036   59f0457f0043bac8594e9f5a3310a709244de980a7b0c206d7d0d95f15455d73