Skip to content

Commit ddc248e

Browse files
committed
add a cmd to find error in _training_successful
1 parent 4e733ec commit ddc248e

File tree

1 file changed

+6
-0
lines changed
  • ramp-engine/ramp_engine/aws

1 file changed

+6
-0
lines changed

ramp-engine/ramp_engine/aws/api.py

+6
Original file line numberDiff line numberDiff line change
@@ -834,6 +834,12 @@ def _training_successful(config, instance_id, submission_name, actual_nb_folds=N
834834

835835
cmd = "find {}|egrep 'fold.*/y_pred_test.npz'|wc -l".format(folder)
836836
nb_test_files = int(_run(config, instance_id, cmd, return_output=True))
837+
838+
cmd = f"find {folder}|egrep 'fold.*/error.txt'|wc -l"
839+
n_error = int(_run(config, instance_id, cmd, return_output=True))
840+
if n_error > 0:
841+
return False
842+
837843
if actual_nb_folds is not None:
838844
return nb_folds == nb_train_files == nb_test_files == actual_nb_folds
839845
else:

0 commit comments

Comments
 (0)