36 lines
869 B
Plaintext
36 lines
869 B
Plaintext
# Environment adapters.
|
|
# ---------------------
|
|
# Atari
|
|
ale_py==0.10.1
|
|
imageio==2.34.2
|
|
opencv-python-headless==4.9.0.80
|
|
|
|
# For testing MuJoCo envs with gymnasium.
|
|
mujoco==3.2.4
|
|
dm_control==1.0.12; python_version < "3.12"
|
|
|
|
# For tests on PettingZoo's multi-agent envs.
|
|
pettingzoo==1.24.3
|
|
pymunk==6.2.1
|
|
tinyscaler==1.2.8
|
|
shimmy==2.0.0
|
|
supersuit==3.9.3
|
|
|
|
# For tests on minigrid.
|
|
minigrid==2.3.1
|
|
tensorflow_estimator
|
|
|
|
# DeepMind's OpenSpiel
|
|
open-spiel==1.4
|
|
|
|
# Requires libtorrent which is unavailable for arm64
|
|
h5py==3.12.1
|
|
|
|
# Requirements for rendering.
|
|
moviepy
|
|
|
|
# For ONNX export tests (policy_inference_after_training examples, --use-onnx-for-inference).
|
|
onnx==1.16.0; sys_platform != 'darwin' or platform_machine != 'arm64'
|
|
onnxruntime==1.18.0; sys_platform != 'darwin' or platform_machine != 'arm64'
|
|
onnxscript; sys_platform != 'darwin' or platform_machine != 'arm64'
|