function set_mpl() {
export MOJO_PYTHON_LIBRARY="$(find $CONDA_PREFIX/lib -iname 'libpython*.[s,d]*' | sort -r | head -n 1)"
}
function setup_mpl_for_python_env() {
# Check that the conda prefix does not contain miniconda
if [[ $CONDA_PREFIX == *miniconda[0-9] ]]; then
echo "You are using the base conda environment. Please activate another environment."
return 1
fi
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
mkdir -p $CONDA_PREFIX/etc/conda/deactivate.d
set_mpl
echo "export MOJO_PYTHON_LIBRARY=$MOJO_PYTHON_LIBRARY" >$CONDA_PREFIX/etc/conda/activate.d/export_mojo.sh
echo "unset MOJO_PYTHON_LIBRARY" >$CONDA_PREFIX/etc/conda/deactivate.d/unset_mojo.sh
}