| Philipp Schrader | 9e1b9bd | 2021-12-28 00:15:12 -0800 | [diff] [blame] | 1 | # After updating this file, run: |
| 2 | # $ bazel run //tools/python:requirements.update | ||||
| Philipp Schrader | 868070a | 2022-09-06 22:51:13 -0700 | [diff] [blame] | 3 | |
| Philipp Schrader | 469a2f2 | 2022-10-29 14:25:58 -0700 | [diff] [blame] | 4 | jinja2 |
| Philipp Schrader | 9e1b9bd | 2021-12-28 00:15:12 -0800 | [diff] [blame] | 5 | matplotlib |
| Philipp Schrader | 469a2f2 | 2022-10-29 14:25:58 -0700 | [diff] [blame] | 6 | mkdocs |
| Philipp Schrader | 9e1b9bd | 2021-12-28 00:15:12 -0800 | [diff] [blame] | 7 | numpy |
| Philipp Schrader | 469a2f2 | 2022-10-29 14:25:58 -0700 | [diff] [blame] | 8 | opencv-python |
| 9 | osqp | ||||
| Philipp Schrader | 868070a | 2022-09-06 22:51:13 -0700 | [diff] [blame] | 10 | pkginfo |
| Austin Schuh | ea5f0a7 | 2024-09-02 15:02:36 -0700 | [diff] [blame] | 11 | pygobject==3.42.2 |
| Philipp Schrader | 868070a | 2022-09-06 22:51:13 -0700 | [diff] [blame] | 12 | requests |
| Philipp Schrader | 9e1b9bd | 2021-12-28 00:15:12 -0800 | [diff] [blame] | 13 | scipy |
| Philipp Schrader | bd39103 | 2023-01-03 00:59:56 -0800 | [diff] [blame] | 14 | shapely |
| James Kuszmaul | b109116 | 2023-12-27 16:16:45 -0800 | [diff] [blame] | 15 | validators |
| Philipp Schrader | b648804 | 2022-11-24 14:35:38 -0800 | [diff] [blame] | 16 | yapf |
| Austin Schuh | 999a19e | 2024-05-04 14:52:39 -0700 | [diff] [blame] | 17 | sympy |
| Austin Schuh | c2f1ca2 | 2024-09-02 13:28:54 -0700 | [diff] [blame] | 18 | absl-py |
| Philipp Schrader | 272e07b | 2024-03-29 17:26:02 -0700 | [diff] [blame] | 19 | pyyaml |
| Philipp Schrader | bd39103 | 2023-01-03 00:59:56 -0800 | [diff] [blame] | 20 | |
| 21 | # TODO(phil): Migrate to absl-py. These are abandoned as far as I can tell. | ||||
| 22 | python-gflags | ||||
| 23 | glog | ||||
| Maxwell Gumley | d26e629 | 2024-04-24 10:45:07 -0600 | [diff] [blame] | 24 | |
| 25 | bokeh | ||||
| 26 | tabulate | ||||
| Austin Schuh | e607e51 | 2024-06-23 22:06:28 -0700 | [diff] [blame] | 27 | |
| Austin Schuh | f8c8892 | 2024-09-21 16:53:58 -0700 | [diff] [blame] | 28 | flask |
| 29 | |||||
| Austin Schuh | dd1f872 | 2024-08-18 00:09:01 -0700 | [diff] [blame] | 30 | casadi>=3.6.6 |
| Austin Schuh | 9327d71 | 2024-09-02 13:49:18 -0700 | [diff] [blame] | 31 | |
| Austin Schuh | f8c8892 | 2024-09-21 16:53:58 -0700 | [diff] [blame] | 32 | # ML libraries |
| Austin Schuh | 9327d71 | 2024-09-02 13:49:18 -0700 | [diff] [blame] | 33 | jax[cuda12] |
| Austin Schuh | f8c8892 | 2024-09-21 16:53:58 -0700 | [diff] [blame] | 34 | jaxtyping |
| Austin Schuh | ea5f0a7 | 2024-09-02 15:02:36 -0700 | [diff] [blame] | 35 | optax |
| 36 | flax | ||||
| 37 | clu | ||||
| 38 | |||||
| 39 | # Pin this so we can actually depend on it. Used by tensorflow/jax. | ||||
| 40 | setuptools==74.0.0 | ||||
| 41 | |||||
| 42 | tensorflow | ||||
| 43 | tensorflow_datasets | ||||
| James Kuszmaul | 3c58132 | 2024-09-14 13:19:26 -0700 | [diff] [blame] | 44 | |
| Austin Schuh | f8c8892 | 2024-09-21 16:53:58 -0700 | [diff] [blame] | 45 | # Experience buffer for reinforcement learning |
| 46 | flashbax | ||||
| justinT21 | a4fa3e2 | 2024-10-12 17:09:50 -0700 | [diff] [blame^] | 47 | |
| 48 | # Experiment tracking | ||||
| 49 | aim | ||||