requirements-ml.txt

#
# This file is autogenerated by pip-compile with Python 3.10
# by the following command:
#
#    pip-compile --extra=ml --output-file=requirements-ml.txt pyproject.toml
#
aiohappyeyeballs==2.4.6
    # via aiohttp
aiohttp==3.11.12
    # via
    #   aiohttp-cors
    #   ray
aiohttp-cors==0.7.0
    # via ray
aiosignal==1.3.2
    # via
    #   aiohttp
    #   ray
annotated-types==0.7.0
    # via pydantic
asttokens==3.0.0
    # via stack-data
async-timeout==5.0.1
    # via
    #   aiohttp
    #   redis
attrs==25.1.0
    # via
    #   aiohttp
    #   flatland-rl (pyproject.toml)
    #   jsonschema
    #   referencing
boto3==1.36.23
    # via flatland-rl (pyproject.toml)
botocore==1.36.23
    # via
    #   boto3
    #   s3transfer
cachetools==5.5.1
    # via google-auth
certifi==2025.1.31
    # via
    #   requests
    #   sentry-sdk
charset-normalizer==3.4.1
    # via requests
click==8.1.8
    # via
    #   flatland-rl (pyproject.toml)
    #   ray
    #   wandb
cloudpickle==3.1.1
    # via
    #   gymnasium
    #   stable-baselines3
colorful==0.5.6
    # via ray
comm==0.2.2
    # via ipywidgets
contourpy==1.3.1
    # via matplotlib
crowdai-api==0.1.22
    # via flatland-rl (pyproject.toml)
cycler==0.12.1
    # via matplotlib
dataclasses==0.6
    # via flatland-rl (pyproject.toml)
decorator==5.1.1
    # via ipython
distlib==0.3.9
    # via virtualenv
dm-tree==0.1.8
    # via
    #   flatland-rl (pyproject.toml)
    #   ray
docker-pycreds==0.4.0
    # via wandb
exceptiongroup==1.2.2
    # via ipython
executing==2.2.0
    # via stack-data
farama-notifications==0.0.4
    # via gymnasium
filelock==3.17.0
    # via
    #   ray
    #   torch
    #   virtualenv
fonttools==4.56.0
    # via matplotlib
frozenlist==1.5.0
    # via
    #   aiohttp
    #   aiosignal
    #   ray
fsspec==2025.2.0
    # via
    #   ray
    #   torch
gitdb==4.0.12
    # via gitpython
gitpython==3.1.44
    # via wandb
google-api-core==2.24.1
    # via opencensus
google-auth==2.38.0
    # via google-api-core
googleapis-common-protos==1.67.0
    # via google-api-core
graphviz==0.20.3
    # via flatland-rl (pyproject.toml)
grpcio==1.70.0
    # via ray
gymnasium==1.0.0
    # via
    #   pettingzoo
    #   ray
    #   stable-baselines3
    #   supersuit
idna==3.10
    # via
    #   requests
    #   yarl
importlib-resources==1.5.0
    # via flatland-rl (pyproject.toml)
ipycanvas==0.13.3
    # via flatland-rl (pyproject.toml)
ipyevents==2.0.2
    # via flatland-rl (pyproject.toml)
ipython==8.32.0
    # via
    #   flatland-rl (pyproject.toml)
    #   ipywidgets
ipywidgets==8.1.5
    # via
    #   flatland-rl (pyproject.toml)
    #   ipycanvas
    #   ipyevents
jedi==0.19.2
    # via ipython
jinja2==3.1.6
    # via torch
jmespath==1.0.1
    # via
    #   boto3
    #   botocore
jsonschema==4.23.0
    # via ray
jsonschema-specifications==2024.10.1
    # via jsonschema
jupyterlab-widgets==3.0.13
    # via ipywidgets
kiwisolver==1.4.8
    # via matplotlib
lxml==5.3.1
    # via svgutils
lz4==4.4.3
    # via ray
markupsafe==3.0.2
    # via jinja2
matplotlib==3.10.0
    # via
    #   flatland-rl (pyproject.toml)
    #   seaborn
    #   stable-baselines3
matplotlib-inline==0.1.7
    # via ipython
mpmath==1.3.0
    # via sympy
msgpack==1.1.0
    # via
    #   flatland-rl (pyproject.toml)
    #   msgpack-numpy
    #   ray
msgpack-numpy==0.4.8
    # via flatland-rl (pyproject.toml)
multidict==6.1.0
    # via
    #   aiohttp
    #   yarl
networkx==3.4.2
    # via
    #   flatland-rl (pyproject.toml)
    #   torch
numpy==1.26.4
    # via
    #   contourpy
    #   flatland-rl (pyproject.toml)
    #   gymnasium
    #   ipycanvas
    #   matplotlib
    #   msgpack-numpy
    #   pandas
    #   pettingzoo
    #   pyarrow
    #   scipy
    #   seaborn
    #   stable-baselines3
    #   supersuit
    #   tensorboardx
    #   tinyscaler
opencensus==0.11.4
    # via ray
opencensus-context==0.1.3
    # via opencensus
ormsgpack==1.7.0
    # via ray
packaging==24.2
    # via
    #   matplotlib
    #   ray
    #   tensorboardx
pandas==2.2.3
    # via
    #   flatland-rl (pyproject.toml)
    #   ray
    #   seaborn
    #   stable-baselines3
parso==0.8.4
    # via jedi
pettingzoo==1.24.3
    # via flatland-rl (pyproject.toml)
pexpect==4.9.0
    # via ipython
pillow==11.1.0
    # via
    #   flatland-rl (pyproject.toml)
    #   ipycanvas
    #   matplotlib
platformdirs==4.3.6
    # via
    #   virtualenv
    #   wandb
prometheus-client==0.21.1
    # via ray
prompt-toolkit==3.0.50
    # via ipython
propcache==0.2.1
    # via
    #   aiohttp
    #   yarl
proto-plus==1.26.0
    # via google-api-core
protobuf==5.29.3
    # via
    #   google-api-core
    #   googleapis-common-protos
    #   proto-plus
    #   ray
    #   tensorboardx
    #   wandb
psutil==7.0.0
    # via wandb
ptyprocess==0.7.0
    # via pexpect
pure-eval==0.2.3
    # via stack-data
py-spy==0.4.0
    # via ray
pyarrow==17.0.0
    # via ray
pyasn1==0.6.1
    # via
    #   pyasn1-modules
    #   rsa
pyasn1-modules==0.4.1
    # via google-auth
pydantic==2.10.6
    # via
    #   ray
    #   wandb
pydantic-core==2.27.2
    # via pydantic
pyglet==2.1.2
    # via flatland-rl (pyproject.toml)
pygments==2.19.1
    # via ipython
pyparsing==3.2.1
    # via matplotlib
python-dateutil==2.9.0.post0
    # via
    #   botocore
    #   matplotlib
    #   pandas
python-gitlab==5.6.0
    # via crowdai-api
pytz==2025.1
    # via pandas
pyyaml==6.0.2
    # via
    #   flatland-rl (pyproject.toml)
    #   ray
    #   wandb
ray[default,rllib]==2.42.1
    # via flatland-rl (pyproject.toml)
recordtype==1.4
    # via flatland-rl (pyproject.toml)
redis==5.2.1
    # via
    #   crowdai-api
    #   flatland-rl (pyproject.toml)
referencing==0.36.2
    # via
    #   jsonschema
    #   jsonschema-specifications
requests==2.32.3
    # via
    #   crowdai-api
    #   google-api-core
    #   python-gitlab
    #   ray
    #   requests-toolbelt
    #   wandb
requests-toolbelt==1.0.0
    # via python-gitlab
rpds-py==0.22.3
    # via
    #   jsonschema
    #   referencing
rsa==4.9
    # via google-auth
s3transfer==0.11.2
    # via boto3
scipy==1.15.2
    # via ray
seaborn==0.13.2
    # via flatland-rl (pyproject.toml)
sentry-sdk==2.22.0
    # via wandb
setproctitle==1.3.4
    # via wandb
six==1.17.0
    # via
    #   docker-pycreds
    #   opencensus
    #   python-dateutil
    #   recordtype
smart-open==7.1.0
    # via ray
smmap==5.0.2
    # via gitdb
stable-baselines3==2.4.1
    # via flatland-rl (pyproject.toml)
stack-data==0.6.3
    # via ipython
supersuit==3.9.3
    # via flatland-rl (pyproject.toml)
svgutils==0.3.4
    # via flatland-rl (pyproject.toml)
sympy==1.13.3
    # via torch
tensorboardx==2.6.2.2
    # via ray
timeout-decorator==0.5.0
    # via flatland-rl (pyproject.toml)
tinyscaler==1.2.8
    # via supersuit
torch==2.2.2
    # via
    #   flatland-rl (pyproject.toml)
    #   stable-baselines3
traitlets==5.14.3
    # via
    #   comm
    #   ipython
    #   ipywidgets
    #   matplotlib-inline
typing-extensions==4.12.2
    # via
    #   gymnasium
    #   ipython
    #   multidict
    #   pydantic
    #   pydantic-core
    #   referencing
    #   torch
    #   wandb
tzdata==2025.1
    # via pandas
urllib3==2.3.0
    # via
    #   botocore
    #   requests
    #   sentry-sdk
virtualenv==20.29.2
    # via ray
wandb==0.19.6
    # via flatland-rl (pyproject.toml)
wcwidth==0.2.13
    # via prompt-toolkit
widgetsnbextension==4.0.13
    # via ipywidgets
wrapt==1.17.2
    # via smart-open
yarl==1.18.3
    # via aiohttp

# The following packages are considered to be unsafe in a requirements file:
# setuptools