|
50 | 50 | False, |
51 | 51 | 'Whether or not to actually run the docker containers. ' |
52 | 52 | 'If False, simply print the docker run commands. ') |
53 | | -flags.DEFINE_enum('tuning_ruleset', |
54 | | - 'external', |
55 | | - enum_values=['external', 'self'], |
56 | | - help='Can be either external of self.') |
| 53 | +flags.DEFINE_enum( |
| 54 | + 'tuning_ruleset', |
| 55 | + 'external', |
| 56 | + enum_values=['external', 'self'], |
| 57 | + help='Can be either external of self.') |
57 | 58 | flags.DEFINE_integer('num_studies', 5, 'Number of studies to run') |
58 | 59 | flags.DEFINE_integer('study_start_index', None, 'Start index for studies.') |
59 | 60 | flags.DEFINE_integer('study_end_index', None, 'End index for studies.') |
|
71 | 72 | 'Path to config containing held-out workloads') |
72 | 73 | flags.DEFINE_string( |
73 | 74 | 'workload_metadata_path', |
74 | | - None, |
| 75 | + None, |
75 | 76 | 'Path to config containing dataset and maximum number of steps per workload.' |
76 | 77 | 'The default values of these are set to the full budgets as determined ' |
77 | 78 | 'via the target-setting procedure. ' |
|
84 | 85 |
|
85 | 86 | FLAGS = flags.FLAGS |
86 | 87 |
|
| 88 | + |
87 | 89 | def read_held_out_workloads(filename): |
88 | 90 | with open(filename, "r") as f: |
89 | 91 | held_out_workloads = json.load(f) |
@@ -195,9 +197,6 @@ def main(_): |
195 | 197 |
|
196 | 198 | command += tuning_ruleset_flags |
197 | 199 |
|
198 | | - |
199 | | - |
200 | | - |
201 | 200 | if not FLAGS.dry_run: |
202 | 201 | print('Running docker container command') |
203 | 202 | print('Container ID: ') |
|
0 commit comments