From 97520d85c9e6faff0711cf6c4331911f10be4464 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Mon, 2 Sep 2024 16:02:05 +0100 Subject: [PATCH 1/3] Update [ghstack-poisoned] --- .github/unittest/linux/scripts/run_all.sh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.github/unittest/linux/scripts/run_all.sh b/.github/unittest/linux/scripts/run_all.sh index 3257adf8c63..3421b11f550 100755 --- a/.github/unittest/linux/scripts/run_all.sh +++ b/.github/unittest/linux/scripts/run_all.sh @@ -189,9 +189,14 @@ export MKL_THREADING_LAYER=GNU export CKPT_BACKEND=torch export MAX_IDLE_COUNT=100 export BATCHED_PIPE_TIMEOUT=60 +export TORCHDYNAMO_INLINE_INBUILT_NN_MODULES=1 pytest test/smoke_test.py -v --durations 200 pytest test/smoke_test_deps.py -v --durations 200 -k 'test_gym or test_dm_control_pixels or test_dm_control or test_tb' + +# Check that benchmarks run +python -m pytest benchmarks + if [ "${CU_VERSION:-}" != cpu ] ; then python .github/unittest/helpers/coverage_run_parallel.py -m pytest test \ --instafail --durations 200 -vv --capture no --ignore test/test_rlhf.py \ From 855d4be2fcdbd7d321ce558f1b8f84d4ec9d16b8 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Mon, 2 Sep 2024 16:56:07 +0100 Subject: [PATCH 2/3] Update [ghstack-poisoned] --- .github/unittest/linux/scripts/environment.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/unittest/linux/scripts/environment.yml b/.github/unittest/linux/scripts/environment.yml index 2234683a497..38d95eb3353 100644 --- a/.github/unittest/linux/scripts/environment.yml +++ b/.github/unittest/linux/scripts/environment.yml @@ -15,6 +15,7 @@ dependencies: - pytest-cov - pytest-mock - pytest-instafail + - pytest-benchmark - pytest-rerunfailures - pytest-timeout - expecttest From 51b013359392218de53f92b8a84dcd688bf3e180 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Tue, 17 Sep 2024 14:17:33 -0700 Subject: [PATCH 3/3] Update [ghstack-poisoned] --- .github/unittest/linux/scripts/run_all.sh | 2 +- .github/unittest/linux_libs/scripts_gym/batch_scripts.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/unittest/linux/scripts/run_all.sh b/.github/unittest/linux/scripts/run_all.sh index 09eefa9f65c..1026b31cf0f 100755 --- a/.github/unittest/linux/scripts/run_all.sh +++ b/.github/unittest/linux/scripts/run_all.sh @@ -88,7 +88,7 @@ conda deactivate conda activate "${env_dir}" echo "installing gymnasium" -pip3 install "gymnasium[ale-py,atari,accept-rom-license]" +pip3 install "gymnasium[atari,accept-rom-license]" pip3 install mo-gymnasium[mujoco] # requires here bc needs mujoco-py pip3 install "mujoco" -U diff --git a/.github/unittest/linux_libs/scripts_gym/batch_scripts.sh b/.github/unittest/linux_libs/scripts_gym/batch_scripts.sh index 9622984a421..a99e4b5a104 100755 --- a/.github/unittest/linux_libs/scripts_gym/batch_scripts.sh +++ b/.github/unittest/linux_libs/scripts_gym/batch_scripts.sh @@ -126,7 +126,7 @@ do conda activate ./cloned_env echo "Testing gym version: ${GYM_VERSION}" - pip3 install 'gymnasium[atari,accept-rom-license,ale-py]'==$GYM_VERSION + pip3 install 'gymnasium[atari,accept-rom-license]'==$GYM_VERSION $DIR/run_test.sh