requirements.txt

#
# This file is autogenerated by pip-compile with Python 3.10
# by the following command:
#
#    pip-compile --extra=dev --extra=launch_jobs --output-file=requirements.txt.new pyproject.toml
#
absl-py==2.1.0
    # via
    #   chex
    #   distrax
    #   dm-env
    #   optax
    #   orbax-checkpoint
    #   rlax
    #   tensorboard
    #   tensorflow-probability
aiosignal==1.3.2
    # via ray
attrs==24.3.0
    # via
    #   jsonschema
    #   referencing
cachetools==5.5.0
    # via google-auth
certifi==2024.12.14
    # via
    #   requests
    #   sentry-sdk
cfgv==3.4.0
    # via pre-commit
charset-normalizer==3.4.1
    # via requests
chex==0.1.88
    # via
    #   distrax
    #   optax
    #   rlax
    #   train-learned-planner (pyproject.toml)
click==8.1.8
    # via
    #   ray
    #   wandb
cloudpickle==3.1.0
    # via
    #   gymnasium
    #   tensorflow-probability
cmdkit==2.7.7
    # via names-generator
contourpy==1.3.1
    # via matplotlib
cycler==0.12.1
    # via matplotlib
databind @ git+https://github.com/rhaps0dy/python-databind.git@merge-fixes#subdirectory=databind
    # via farconf
decorator==4.4.2
    # via
    #   moviepy
    #   tensorflow-probability
deprecated==1.2.15
    # via databind
distlib==0.3.9
    # via virtualenv
distrax==0.1.5
    # via rlax
dm-env==1.6
    # via rlax
dm-tree==0.1.8
    # via
    #   dm-env
    #   tensorflow-probability
docker-pycreds==0.4.0
    # via wandb
etils[epath,epy]==1.11.0
    # via orbax-checkpoint
exceptiongroup==1.2.2
    # via pytest
farama-notifications==0.0.4
    # via gymnasium
farconf @ git+https://github.com/AlignmentResearch/farconf.git
    # via train-learned-planner (pyproject.toml)
filelock==3.16.1
    # via
    #   huggingface-hub
    #   ray
    #   virtualenv
flax==0.8.5
    # via train-learned-planner (pyproject.toml)
fonttools==4.55.3
    # via matplotlib
frozenlist==1.5.0
    # via
    #   aiosignal
    #   ray
fsspec==2024.12.0
    # via
    #   etils
    #   huggingface-hub
    #   ray
gast==0.6.0
    # via tensorflow-probability
gitdb==4.0.11
    # via gitpython
gitpython==3.1.43
    # via
    #   train-learned-planner (pyproject.toml)
    #   wandb
google-auth==2.37.0
    # via
    #   google-auth-oauthlib
    #   tensorboard
google-auth-oauthlib==1.0.0
    # via tensorboard
grpcio==1.68.1
    # via tensorboard
gymnasium==0.29.1
    # via train-learned-planner (pyproject.toml)
huggingface-hub==0.23.5
    # via train-learned-planner (pyproject.toml)
humanize==4.11.0
    # via orbax-checkpoint
identify==2.6.4
    # via pre-commit
idna==3.10
    # via requests
imageio==2.36.1
    # via moviepy
imageio-ffmpeg==0.5.1
    # via moviepy
importlib-resources==6.4.5
    # via etils
iniconfig==2.0.0
    # via pytest
# DISABLED jax==0.4.38
    # via
    #   chex
    #   distrax
    #   flax
    #   optax
    #   orbax-checkpoint
    #   rlax
# DISABLED jaxlib==0.4.38
    # via
    #   chex
    #   distrax
    #   jax
    #   optax
    #   rlax
jsonschema==4.23.0
    # via ray
jsonschema-specifications==2024.10.1
    # via jsonschema
kiwisolver==1.4.8
    # via matplotlib
markdown==3.7
    # via tensorboard
markdown-it-py==3.0.0
    # via rich
markupsafe==3.0.2
    # via werkzeug
matplotlib==3.9.4
    # via train-learned-planner (pyproject.toml)
mdurl==0.1.2
    # via markdown-it-py
ml-dtypes==0.5.0
    # via
    #   jax
    #   jaxlib
    #   tensorstore
moviepy==1.0.3
    # via train-learned-planner (pyproject.toml)
msgpack==1.1.0
    # via
    #   flax
    #   orbax-checkpoint
    #   ray
names-generator==0.1.0
    # via train-learned-planner (pyproject.toml)
nest-asyncio==1.6.0
    # via orbax-checkpoint
nodeenv==1.9.1
    # via
    #   pre-commit
    #   pyright
nr-date==2.1.0
    # via databind
nr-stream==1.1.5
    # via databind
numpy==1.26.4
    # via
    #   chex
    #   contourpy
    #   distrax
    #   dm-env
    #   flax
    #   gymnasium
    #   imageio
    #   jax
    #   jaxlib
    #   matplotlib
    #   ml-dtypes
    #   moviepy
    #   opencv-python
    #   optax
    #   orbax-checkpoint
    #   pandas
    #   rlax
    #   scipy
    #   tensorboard
    #   tensorboardx
    #   tensorflow-probability
    #   tensorstore
oauthlib==3.2.2
    # via requests-oauthlib
opencv-python==4.10.0.84
    # via train-learned-planner (pyproject.toml)
opt-einsum==3.4.0
    # via jax
optax==0.1.9
    # via
    #   flax
    #   train-learned-planner (pyproject.toml)
orbax-checkpoint==0.11.0
    # via flax
packaging==24.2
    # via
    #   huggingface-hub
    #   matplotlib
    #   pytest
    #   ray
    #   tensorboardx
pandas==2.2.3
    # via ray
pillow==11.0.0
    # via
    #   imageio
    #   matplotlib
platformdirs==4.3.6
    # via
    #   virtualenv
    #   wandb
pluggy==1.5.0
    # via pytest
pre-commit==3.6.2
    # via train-learned-planner (pyproject.toml)
proglog==0.1.10
    # via moviepy
protobuf==5.29.2
    # via
    #   orbax-checkpoint
    #   ray
    #   tensorboard
    #   tensorboardx
    #   wandb
psutil==6.1.1
    # via wandb
pyarrow==18.1.0
    # via ray
pyasn1==0.6.1
    # via
    #   pyasn1-modules
    #   rsa
pyasn1-modules==0.4.1
    # via google-auth
pygments==2.18.0
    # via rich
pyparsing==3.2.0
    # via matplotlib
pyright==1.1.391
    # via train-learned-planner (pyproject.toml)
pytest==8.1.2
    # via train-learned-planner (pyproject.toml)
python-dateutil==2.9.0.post0
    # via
    #   matplotlib
    #   pandas
pytz==2024.2
    # via pandas
pyyaml==6.0.2
    # via
    #   farconf
    #   flax
    #   huggingface-hub
    #   orbax-checkpoint
    #   pre-commit
    #   ray
    #   wandb
ray[tune]==2.40.0
    # via train-learned-planner (pyproject.toml)
referencing==0.35.1
    # via
    #   jsonschema
    #   jsonschema-specifications
requests==2.32.3
    # via
    #   huggingface-hub
    #   moviepy
    #   ray
    #   requests-oauthlib
    #   tensorboard
    #   wandb
requests-oauthlib==2.0.0
    # via google-auth-oauthlib
rich==13.9.4
    # via
    #   flax
    #   train-learned-planner (pyproject.toml)
rlax==0.1.6
    # via train-learned-planner (pyproject.toml)
rpds-py==0.22.3
    # via
    #   jsonschema
    #   referencing
rsa==4.9
    # via google-auth
ruff==0.1.15
    # via train-learned-planner (pyproject.toml)
scipy==1.14.1
    # via
    #   jax
    #   jaxlib
sentry-sdk==2.19.2
    # via wandb
setproctitle==1.3.4
    # via wandb
simplejson==3.19.3
    # via orbax-checkpoint
six==1.17.0
    # via
    #   docker-pycreds
    #   python-dateutil
    #   tensorflow-probability
smmap==5.0.1
    # via gitdb
tensorboard==2.12.3
    # via train-learned-planner (pyproject.toml)
tensorboard-data-server==0.7.2
    # via tensorboard
tensorboardx==2.6.2.2
    # via
    #   ray
    #   train-learned-planner (pyproject.toml)
tensorflow-probability==0.25.0
    # via distrax
tensorstore==0.1.71
    # via
    #   flax
    #   orbax-checkpoint
toml==0.10.2
    # via cmdkit
tomli==2.2.1
    # via pytest
toolz==1.0.0
    # via chex
tqdm==4.67.1
    # via
    #   huggingface-hub
    #   moviepy
    #   proglog
typeapi==2.2.3
    # via
    #   databind
    #   farconf
typing-extensions==4.12.2
    # via
    #   chex
    #   databind
    #   etils
    #   flax
    #   gymnasium
    #   huggingface-hub
    #   orbax-checkpoint
    #   pyright
    #   rich
    #   typeapi
tzdata==2024.2
    # via pandas
urllib3==2.3.0
    # via
    #   requests
    #   sentry-sdk
virtualenv==20.28.0
    # via pre-commit
wandb==0.17.9
    # via train-learned-planner (pyproject.toml)
werkzeug==3.1.3
    # via tensorboard
wheel==0.45.1
    # via tensorboard
wrapt==1.17.0
    # via deprecated
zipp==3.21.0
    # via etils

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