Generates requirements.txt file with the Acme's dependencies. It is used by Launchpad GCP runtime to generate Acme requirements to be installed inside the docker image. Acme itself is not installed from pypi, but instead sources are copied over to reflect any local changes made to the codeb
(path=None)
| 87 | |
| 88 | |
| 89 | def generate_requirements_file(path=None): |
| 90 | """Generates requirements.txt file with the Acme's dependencies. |
| 91 | |
| 92 | It is used by Launchpad GCP runtime to generate Acme requirements to be |
| 93 | installed inside the docker image. Acme itself is not installed from pypi, |
| 94 | but instead sources are copied over to reflect any local changes made to |
| 95 | the codebase. |
| 96 | |
| 97 | Args: |
| 98 | path: path to the requirements.txt file to generate. |
| 99 | """ |
| 100 | if not path: |
| 101 | path = os.path.join(os.path.dirname(__file__), 'acme/requirements.txt') |
| 102 | with open(path, 'w') as f: |
| 103 | for package in set(core_requirements + jax_requirements + tf_requirements + |
| 104 | envs_requirements): |
| 105 | f.write(f'{package}\n') |
| 106 | |
| 107 | |
| 108 | long_description = """Acme is a library of reinforcement learning (RL) agents |