Merge branch 'main' of https://github.com/pytorch/rl into fix/multisync

Author: ParamThakkar123

.github/unittest/linux_libs/scripts_chess/run_test.sh

@@ -22,7 +22,7 @@ conda deactivate && conda activate ./env
 # this workflow only tests the libs
 python -c "import chess"
 
-python .github/unittest/helpers/coverage_run_parallel.py -m pytest test/test_env.py --instafail -v --durations 200 --capture no -k TestChessEnv --error-for-skips --runslow
+python .github/unittest/helpers/coverage_run_parallel.py -m pytest test/test_envs.py --instafail -v --durations 200 --capture no -k TestChessEnv --error-for-skips --runslow
 
 coverage combine -q
 coverage xml -i

.github/unittest/linux_libs/scripts_jumanji/environment.yml

@@ -21,3 +21,4 @@ dependencies:
     - hydra-core
     - jumanji
     - gymnasium<1.0.0
+    - requests

.github/unittest/linux_libs/scripts_jumanji/run_test.sh

@@ -20,6 +20,10 @@ lib_dir="${env_dir}/lib"
 # solves ImportError: /lib64/libstdc++.so.6: version `GLIBCXX_3.4.21' not found
 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$lib_dir
 export MKL_THREADING_LAYER=GNU
+# Limit threading to avoid resource exhaustion (pthread_create failures)
+export OMP_NUM_THREADS=1
+export MKL_NUM_THREADS=1
+export XLA_FLAGS="--xla_force_host_platform_device_count=1"
 # more logging
 export MAGNUM_LOG=verbose MAGNUM_GPU_VALIDATION=ON
 
@@ -29,6 +33,6 @@ export MAGNUM_LOG=verbose MAGNUM_GPU_VALIDATION=ON
 # this workflow only tests the libs
 python -c "import jumanji"
 
-python .github/unittest/helpers/coverage_run_parallel.py -m pytest test/test_libs.py --instafail -v --durations 200 --capture no -k TestJumanji --error-for-skips --runslow
+python .github/unittest/helpers/coverage_run_parallel.py -m pytest test/test_libs.py --instafail -v --durations 200 --capture no -k "TestJumanji and not test_jumanji_batch_unlocked" --error-for-skips --runslow
 coverage combine -q
 coverage xml -i