From 8850da42a6b5218670d07e5ebd5638d7dd4963c6 Mon Sep 17 00:00:00 2001 From: Augustin Zidek Date: Mon, 11 May 2026 01:48:25 -0700 Subject: [PATCH] Validate that the output_dir is always set and add a missing argument in overload PiperOrigin-RevId: 913567460 Change-Id: I3b393fdc12819265a6916cb2c735964d25b3e375 --- run_alphafold.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/run_alphafold.py b/run_alphafold.py index 4b8dd70..ef5deb3 100644 --- a/run_alphafold.py +++ b/run_alphafold.py @@ -707,6 +707,7 @@ def process_fold_input( ref_max_modified_date: datetime.date | None = None, conformer_max_iterations: int | None = None, resolve_msa_overlaps: bool = True, + fix_standalone_glycans: bool = False, force_output_dir: bool = False, compress_large_output_files: bool = False, ) -> folding_input.Input: @@ -724,6 +725,7 @@ def process_fold_input( ref_max_modified_date: datetime.date | None = None, conformer_max_iterations: int | None = None, resolve_msa_overlaps: bool = True, + fix_standalone_glycans: bool = False, force_output_dir: bool = False, compress_large_output_files: bool = False, ) -> Sequence[ResultsForSeed]: @@ -875,6 +877,9 @@ def main(_): 'Exactly one of --json_path or --input_dir must be specified.' ) + if _OUTPUT_DIR.value is None: + raise ValueError('Output directory must be specified with --output_dir.') + # Make sure we can create the output directory before running anything. try: os.makedirs(_OUTPUT_DIR.value, exist_ok=True)