Skip to content

Commit a70448e

Browse files
committed
fix
1 parent ccd9fbb commit a70448e

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

scoring/run_workloads.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,10 @@ def main(_):
128128
if FLAGS.hparam_end_index:
129129
hparam_end_index_flag = f'--hparam_end_index {FLAGS.hparam_end_index} '
130130
study_start_index = FLAGS.study_start_index if FLAGS.study_start_index else 0
131-
study_end_index = FLAGS.study_end_index if FLAGS.study_end_index else num_studies - 1
131+
if FLAGS.study_end_index is not None:
132+
study_end_index = FLAGS.study_end_index
133+
else:
134+
study_end_index = num_studies - 1
132135
submission_id = FLAGS.submission_id
133136
rng_seed = FLAGS.seed
134137

@@ -176,7 +179,7 @@ def main(_):
176179
mount_repo_flag = ''
177180
if FLAGS.local:
178181
mount_repo_flag = '-v $HOME/algorithmic-efficiency:/algorithmic-efficiency '
179-
command = ('docker run -t -d -v $HOME/data/:/data/ '
182+
command = ('docker run -t -d -v /home/kasimbeg/data/:/data/ '
180183
'-v $HOME/experiment_runs/:/experiment_runs '
181184
'-v $HOME/experiment_runs/logs:/logs '
182185
f'{mount_repo_flag}'
@@ -232,4 +235,4 @@ def main(_):
232235

233236
if __name__ == '__main__':
234237
flags.mark_flag_as_required('workload_metadata_path')
235-
app.run(main)
238+
app.run(main)

0 commit comments

Comments
 (0)