# 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'