[CI] Fix CI issues #1498

Job	Run time
unittests-pettingzoo / linux-job	54s
unittests-d4rl (3.9, 12.1) / linux-job	59s
unittests-envpool (3.9, 12.1) / linux-job	56s
unittests-gendgrl (3.9, 12.1) / linux-job	53s
unittests-brax (3.11, 12.1) / linux-job	55s
unittests-atari-dqn (3.9, 12.1) / linux-job	57s
unittests-gym (3.9, 12.1) / linux-job	1m 0s
unittests-minari (3.9, 12.1) / linux-job	50s
unittests-jumanji (3.9, 12.1) / linux-job	48s
unittests-openx (3.9, 12.1) / linux-job	57s
unittests-robohive (3.9, 12.1) / linux-job	56s
unittests-sklearn (3.9, 12.1) / linux-job	54s
unittests-roboset (3.9, 12.1) / linux-job	51s
unittests-smacv2 (3.9, 12.1) / linux-job	55s
unittests-vd4rl (3.9, 12.1) / linux-job	45s
unittests-vmas (3.9, 12.1) / linux-job	48s
	14m 18s

Provide feedback