-r requirements_test.txt

cloudpickle
filelock

Pillow
polars
# Version 2.0.0 removes moviepy.editor, used by TensorBoard in some tests
# to log videos.
moviepy~=1.0
imageio[ffmpeg]
matplotlib
soundfile
rdkit
plotly
bokeh

tqdm
tenacity

ipython
jupyter
ipykernel
nbclient

tensorboard
tensorboardx
tensorflow~=2.20; sys_platform != 'darwin'
numpy<2.4.0  # tensorflow==2.20.0 uses deprecated args removed in numpy==2.4.0

scikit-learn
torch
torchvision
jax[cpu]
lightning

pyarrow
kfp
metaflow
xgboost
lightgbm
urllib3
docker
catboost
openai
gymnasium
stable_baselines3
dspy

responses

# `ariadne-codegen` is a dev-only dependency used for GraphQL codegen.
ariadne-codegen

.[launch]
.[sweeps] ; sys_platform != 'darwin' or (sys_platform == 'darwin' and platform.machine != 'arm64')

# Sandbox tests need the upstream client, but it only supports Python 3.11+.
cwsandbox[cli]>=0.20.0; python_version >= '3.11'
