From 215b0dadfa79af10cebf9e9c2753239c3b4e087e Mon Sep 17 00:00:00 2001 From: Aidan Holland Date: Sun, 15 Jan 2023 10:38:57 -0500 Subject: [PATCH] fix: add fallback for step metrics This was throwing errors locally. --- openai/wandb_logger.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)