#!/bin/bash
set -ex

# Install mlflow (assuming the repository root is mounted to the working directory)
if [ "$INSTALL_MLFLOW_FROM_REPO" = "true" ]; then
  uv pip install --system --no-deps -e .
fi

# For Microsoft SQL server, wait until the database is up and running
if [[ $MLFLOW_TRACKING_URI == mssql* ]]; then
  ./tests/db/init-mssql-db.sh
fi

# Execute the command
exec "$@"