Fix validation of runner config options (google#1771)

This PR modifies the config file validation to allow to specify `runner_machine_type` and `runner_num_cpu_cores`.
ardier · Mar 15, 2023 · 653da70 · 653da70
1 parent 4883508
commit 653da70
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/experiment/run_experiment.py b/experiment/run_experiment.py
@@ -181,6 +181,10 @@ def read_and_validate_experiment_config(config_filename: str) -> Dict:
             Requirement(False, bool, False, ''),
         'preemptible_runners':
             Requirement(False, bool, False, ''),
+        'runner_machine_type':
+            Requirement(False, str, True, ''),
+        'runner_num_cpu_cores':
+            Requirement(False, int, False, ''),
     }
 
     all_params_valid = _validate_config_parameters(config, config_requirements)