amend

Author: vmoens

.github/unittest/linux/scripts/environment.yml

@@ -0,0 +1,32 @@
+pip
+protobuf
+hypothesis
+future
+cloudpickle
+pygame
+moviepy<2.0.0
+tqdm
+pytest
+pytest-cov
+pytest-mock
+pytest-instafail
+pytest-rerunfailures
+pytest-timeout
+pytest-asyncio
+expecttest
+pybind11[global]
+pyyaml
+scipy
+hydra-core
+tensorboard
+imageio==2.26.0
+wandb
+dm_control
+mujoco<3.3.6
+mlflow
+av
+coverage
+ray
+transformers
+ninja
+timm

.github/unittest/linux/scripts/requirements.txt

@@ -27,7 +27,7 @@ if [[ $OSTYPE != 'darwin'* ]]; then
     apt-get upgrade -y libstdc++6
     apt-get dist-upgrade -y
   else
-    apt-get install -y g++ gcc
+    apt-get install -y g++ gcc cmake
   fi
 
 fi
@@ -79,12 +79,13 @@ export LAZY_LEGACY_OP=False
 export RL_LOGGING_LEVEL=DEBUG
 export TOKENIZERS_PARALLELISM=true
 
-# Install dependencies from environment.yml using uv
-uv pip install hypothesis future cloudpickle pygame "moviepy<2.0.0" tqdm \
-  pytest pytest-cov pytest-mock pytest-instafail pytest-rerunfailures \
-  pytest-timeout pytest-asyncio expecttest "pybind11[global]" pyyaml scipy \
-  hydra-core tensorboard "imageio==2.26.0" wandb dm_control "mujoco<3.3.6" \
-  mlflow av coverage ray transformers ninja timm protobuf setuptools
+# Install build dependencies FIRST (required for C++ extensions)
+printf "* Installing build dependencies\n"
+uv pip install setuptools wheel ninja "pybind11[global]"
+
+# Install dependencies from requirements.txt
+printf "* Installing dependencies from requirements.txt\n"
+uv pip install -r "${this_dir}/requirements.txt"
 
 # Install pip for compatibility with packages that expect it
 uv pip install pip

.github/unittest/linux/scripts/run_all.sh

@@ -0,0 +1,29 @@
+pip
+protobuf
+hypothesis
+future
+cloudpickle
+pygame
+moviepy<2.0.0
+tqdm
+pytest
+pytest-cov
+pytest-mock
+pytest-instafail
+pytest-rerunfailures
+pytest-asyncio
+expecttest
+pybind11[global]
+pyyaml
+scipy
+hydra-core
+tensorboard
+imageio==2.26.0
+wandb
+dm_control
+mujoco<3.3.6
+mlflow
+av
+coverage
+ray
+virtualenv

.github/unittest/linux_distributed/scripts/environment.yml

@@ -9,7 +9,7 @@ set -e
 
 this_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 # Avoid error: "fatal: unsafe repository"
-apt-get update && apt-get install -y git curl wget gcc g++
+apt-get update && apt-get install -y git curl wget gcc g++ cmake
 git config --global --add safe.directory '*'
 root_dir="$(git rev-parse --show-toplevel)"
 env_dir="${root_dir}/.venv"

.github/unittest/linux_distributed/scripts/requirements.txt

@@ -14,7 +14,7 @@ source "${root_dir}/.venv/bin/activate"
 
 # Install build dependencies EARLY (required for --no-build-isolation)
 printf "* Installing build dependencies\n"
-uv pip install setuptools ninja "pybind11[global]"
+uv pip install setuptools wheel ninja "pybind11[global]"
 
 if [ "${CU_VERSION:-}" == cpu ] ; then
     version="cpu"

.github/unittest/linux_distributed/scripts/setup_env.sh

@@ -0,0 +1,21 @@
+pip
+gsutil
+hypothesis
+future
+cloudpickle
+pytest
+pytest-cov
+pytest-mock
+pytest-instafail
+pytest-rerunfailures
+pytest-error-for-skips
+pytest-asyncio
+expecttest
+pybind11[global]
+pyyaml
+scipy
+hydra-core
+tqdm
+h5py
+datasets
+pillow