| """Folding studio CLI.""" | |
| import typer | |
| from folding_studio.commands.experiment import app as experiment_app | |
| from folding_studio.commands.key import app as key_app | |
| from folding_studio.commands.msa import app as msa_app | |
| from folding_studio.commands.predict import ( | |
| af2, | |
| boltz, | |
| chai, | |
| openfold, | |
| protenix, | |
| soloseq, | |
| ) | |
| app = typer.Typer( | |
| no_args_is_help=True, context_settings={"help_option_names": ["-h", "--help"]} | |
| ) | |
| app.add_typer(experiment_app, name="experiment") | |
| app.add_typer(msa_app, name="msa") | |
| app.add_typer(key_app, name="key") | |
| predict_app = typer.Typer(no_args_is_help=True, help="Submit folding jobs") | |
| predict_app.command()(af2) | |
| predict_app.command()(openfold) | |
| predict_app.command()(boltz) | |
| predict_app.command()(chai) | |
| predict_app.command()(protenix) | |
| predict_app.command()(soloseq) | |
| app.add_typer(predict_app, name="predict") | |
| if __name__ == "__main__": | |
| app() | |