use uv

Author: vmoens

.github/unittest/linux_libs/scripts_procgen/environment.yml

@@ -7,7 +7,8 @@ unset PYTORCH_VERSION
 
 set -e
 
-# Note: This script is sourced by run_all.sh, so the environment is already active
+# Activate the virtual environment
+source ./env/bin/activate
 
 if [ "${CU_VERSION:-}" == cpu ] ; then
     version="cpu"

.github/unittest/linux_libs/scripts_procgen/install.sh

@@ -2,5 +2,5 @@
 
 set -e
 
-eval "$(./conda/bin/conda shell.bash hook)"
-conda activate ./env
+# Activate the virtual environment
+source ./env/bin/activate

.github/unittest/linux_libs/scripts_procgen/post_process.sh

@@ -2,8 +2,8 @@
 
 set -e
 
-eval "$(./conda/bin/conda shell.bash hook)"
-conda activate ./env
+# Activate the virtual environment
+source ./env/bin/activate
 
 apt-get update && apt-get install -y git wget cmake
 
@@ -17,7 +17,7 @@ root_dir="$(git rev-parse --show-toplevel)"
 env_dir="${root_dir}/env"
 lib_dir="${env_dir}/lib"
 
-conda deactivate && conda activate ./env
+deactivate 2>/dev/null || true && source ./env/bin/activate
 
 # this workflow only tests the libs
 python -c "import procgen"

.github/unittest/linux_libs/scripts_procgen/run_test.sh

@@ -33,37 +33,32 @@ apt-get upgrade -y libstdc++6
 
 this_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 root_dir="$(git rev-parse --show-toplevel)"
-conda_dir="${root_dir}/conda"
 env_dir="${root_dir}/env"
 
 cd "${root_dir}"
 
-case "$(uname -s)" in
-    Darwin*) os=MacOSX;;
-    *) os=Linux
-esac
-
-# 1. Install conda at ./conda
-if [ ! -d "${conda_dir}" ]; then
-    printf "* Installing conda\n"
-    wget -O miniconda.sh "http://repo.continuum.io/miniconda/Miniconda3-latest-${os}-x86_64.sh"
-    bash ./miniconda.sh -b -f -p "${conda_dir}"
+# Install uv if not already installed
+if ! command -v uv &> /dev/null; then
+    printf "* Installing uv\n"
+    curl -LsSf https://astral.sh/uv/install.sh | sh
+    export PATH="$HOME/.local/bin:$PATH"
 fi
-eval "$(${conda_dir}/bin/conda shell.bash hook)"
 
-# 2. Create test environment at ./env
+# Create virtual environment using uv
 printf "python: ${PYTHON_VERSION}\n"
 if [ ! -d "${env_dir}" ]; then
-    printf "* Creating a test environment\n"
-    conda create --prefix "${env_dir}" -y python="$PYTHON_VERSION"
+    printf "* Creating a test environment with uv\n"
+    uv venv "${env_dir}" --python "${PYTHON_VERSION}"
 fi
-conda activate "${env_dir}"
 
-# 3. Install Conda dependencies
-printf "* Installing dependencies (except PyTorch)\n"
-echo "  - python=${PYTHON_VERSION}" >> "${this_dir}/environment.yml"
-cat "${this_dir}/environment.yml"
+# Activate the virtual environment
+source "${env_dir}/bin/activate"
 
-pip3 install pip --upgrade
+# Upgrade pip
+uv pip install --upgrade pip
 
-conda env update --file "${this_dir}/environment.yml" --prune
+# Install dependencies from requirements.txt
+printf "* Installing dependencies (except PyTorch)\n"
+if [ -f "${this_dir}/requirements.txt" ]; then
+    uv pip install -r "${this_dir}/requirements.txt"
+fi