Allow specifying runner_memory" (google#1786)

ardier · Mar 17, 2023 · 6d20876 · 6d20876
1 parent 6291809
commit 6d20876
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/experiment/run_experiment.py b/experiment/run_experiment.py
@@ -185,6 +185,8 @@ def read_and_validate_experiment_config(config_filename: str) -> Dict:
             Requirement(False, str, True, ''),
         'runner_num_cpu_cores':
             Requirement(False, int, False, ''),
+        'runner_memory':
+            Requirement(False, str, False, ''),
     }
 
     all_params_valid = _validate_config_parameters(config, config_requirements)