vanishingradient commited on
Commit
a48afd7
·
verified ·
1 Parent(s): e819bd7

Add kaggle/train_grpo.py

Browse files
Files changed (1) hide show
  1. kaggle/train_grpo.py +1 -1
kaggle/train_grpo.py CHANGED
@@ -394,7 +394,6 @@ def run_heldout_agent_eval(
394
  "resolution": "ERROR",
395
  }
396
  )
397
- _vlog("heldout_eval.done", n=len(rows))
398
  finally:
399
  try:
400
  env._end_trace()
@@ -404,6 +403,7 @@ def run_heldout_agent_eval(
404
  env._cleanup()
405
  except Exception:
406
  pass
 
407
  return rows
408
 
409
 
 
394
  "resolution": "ERROR",
395
  }
396
  )
 
397
  finally:
398
  try:
399
  env._end_trace()
 
403
  env._cleanup()
404
  except Exception:
405
  pass
406
+ _vlog("heldout_eval.done", n=len(rows))
407
  return rows
408
 
409