diff --git a/main.py b/main.py index 4a58fab..2c4a449 100644 --- a/main.py +++ b/main.py @@ -86,7 +86,7 @@ def log_train_outcome(outcome, diff_in_values, trained_eps = 0, log_path = os.pa 'sum': sum(outcome), 'mean': sum(outcome) / len(outcome), 'time': int(time.time()), - 'average_diff_in_vals': diff_in_values + 'average_diff_in_vals': diff_in_values, 'commit': commit }