fix: save task end times
@@ -1254,7 +1254,7 @@ RETURNING id`, trial); err != nil {
func (db *PgDB) TrialByID(id int) (*model.Trial, error) {
var trial model.Trial
err := db.query(`
-SELECT id, request_id, experiment_id, state, start_time, end_time,
+SELECT id, task_id, request_id, experiment_id, state, start_time, end_time,
hparams, warm_start_checkpoint_id, seed
FROM trials
WHERE id = $1`, &trial, id)
@@ -1267,7 +1267,7 @@ func (db *PgDB) TrialByExperimentAndRequestID(
) (*model.Trial, error) {
var trial model.Trial
err := db.query(`
-SELECT id, request_id, experiment_id, state, start_time,
+SELECT id, task_id, request_id, experiment_id, state, start_time,
end_time, hparams, warm_start_checkpoint_id, seed
FROM trials
WHERE experiment_id = $1 AND request_id = $2`, &trial, experimentID, requestID)
