diff --git a/openai/wandb_logger.py b/openai/wandb_logger.py index ba650d1fe4..2223b6d742 100644 --- a/openai/wandb_logger.py +++ b/openai/wandb_logger.py @@ -159,7 +159,7 @@ def _log_fine_tune( df_results = pd.read_csv(io.StringIO(results)) for _, row in df_results.iterrows(): metrics = {k: v for k, v in row.items() if not np.isnan(v)} - step = metrics.pop("step") + step = metrics.pop("step", None) if step is not None: step = int(step) wandb.log(metrics, step=step)