1  #!/usr/bin/env bash
 2  
 3  function error_handling() {
 4    echo "Error occurred in dev-env-setup.sh script at line: ${1}"
 5    echo "Line exited with status: ${2}"
 6  }
 7  
 8  trap 'error_handling ${LINENO} $?' ERR
 9  
10  set -o errexit
11  set -o errtrace
12  set -o pipefail
13  shopt -s inherit_errexit
14  set -x
15  
16  err=0
17  
18  REPO_ROOT=$(git rev-parse --show-toplevel)
19  export REPO_ROOT
20  
21  # Run the installation of the environment
22  DEV_DIR=$REPO_ROOT/.venvs/mlflow-dev
23  
24  "$REPO_ROOT"/dev/dev-env-setup.sh -d "$DEV_DIR"
25  
26  source "$DEV_DIR/bin/activate"
27  
28  # Check that packages are installed
29  
30  SKLEARN_VER=$(pip freeze | grep "scikit-learn")
31  
32  if [ -z "$SKLEARN_VER" ]; then
33    err=$((err + 1))
34  fi
35  
36  min_py_version="3.10"
37  installed_py_version=$(python --version | grep -E -o "([0-9]{1,}\.[0-9]{1,})")
38  
39  # shellcheck disable=SC2053
40  if [[ $min_py_version != $installed_py_version ]]; then
41    err=$((err + 1))
42  fi
43  
44  test $err = 0