wehub-resource-sync
84 lines
1.5 KiB
Bash
Executable File
84 lines
1.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -ex
|
|
|
|
function retry-with-backoff() {
|
|
for BACKOFF in 0 1 2; do
|
|
sleep $BACKOFF
|
|
if "$@"; then
|
|
return 0
|
|
fi
|
|
done
|
|
return 1
|
|
}
|
|
|
|
while :
|
|
do
|
|
case "$1" in
|
|
# Install skinny dependencies
|
|
--skinny)
|
|
SKINNY="true"
|
|
shift
|
|
;;
|
|
# Install ML dependencies
|
|
--ml)
|
|
ML="true"
|
|
shift
|
|
;;
|
|
--)
|
|
shift
|
|
break
|
|
;;
|
|
-*)
|
|
echo "Error: unknown option: $1" >&2
|
|
exit 1
|
|
;;
|
|
*)
|
|
break
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# Cleanup apt repository to make room for tests.
|
|
sudo apt clean
|
|
df -h
|
|
|
|
python --version
|
|
uv pip install --system --upgrade setuptools wheel
|
|
uv --version
|
|
|
|
if [[ "$SKINNY" == "true" ]]; then
|
|
uv pip install --system ./libs/skinny
|
|
else
|
|
uv pip install --system .[extras,gateway,mcp] --upgrade
|
|
fi
|
|
|
|
req_files=""
|
|
# Install Python test dependencies only if we're running Python tests
|
|
if [[ "$ML" == "true" ]]; then
|
|
req_files+=" -r requirements/extra-ml-requirements.txt"
|
|
fi
|
|
|
|
if [[ "$SKINNY" == "true" ]]; then
|
|
req_files+=" -r requirements/skinny-test-requirements.txt"
|
|
else
|
|
req_files+=" -r requirements/test-requirements.txt"
|
|
fi
|
|
|
|
if [[ ! -z $req_files ]]; then
|
|
retry-with-backoff uv pip install --system $req_files
|
|
fi
|
|
|
|
# Install `mlflow-test-plugin`
|
|
uv pip install --system tests/resources/mlflow-test-plugin
|
|
|
|
# Print current environment info
|
|
uv pip install --system aiohttp
|
|
which mlflow
|
|
|
|
# Print mlflow version
|
|
mlflow --version
|
|
|
|
# Turn off trace output & exit-on-errors
|
|
set +ex
|