Files
mlflow--mlflow/dev/test-dev-env-setup.sh
2026-07-13 13:22:34 +08:00

45 lines
868 B
Bash
Executable File

#!/usr/bin/env bash
function error_handling() {
echo "Error occurred in dev-env-setup.sh script at line: ${1}"
echo "Line exited with status: ${2}"
}
trap 'error_handling ${LINENO} $?' ERR
set -o errexit
set -o errtrace
set -o pipefail
shopt -s inherit_errexit
set -x
err=0
REPO_ROOT=$(git rev-parse --show-toplevel)
export REPO_ROOT
# Run the installation of the environment
DEV_DIR=$REPO_ROOT/.venvs/mlflow-dev
"$REPO_ROOT"/dev/dev-env-setup.sh -d "$DEV_DIR"
source "$DEV_DIR/bin/activate"
# Check that packages are installed
SKLEARN_VER=$(pip freeze | grep "scikit-learn")
if [ -z "$SKLEARN_VER" ]; then
err=$((err + 1))
fi
min_py_version="3.10"
installed_py_version=$(python --version | grep -E -o "([0-9]{1,}\.[0-9]{1,})")
# shellcheck disable=SC2053
if [[ $min_py_version != $installed_py_version ]]; then
err=$((err + 1))
fi
test $err = 0