Skip to content

Commit ce5bf57

Browse files
authored
Fix undefined trackers property (#224)
This bug was introduced by #209, which changed the `trackers` config property to `tracker` but didn't update this use case.
1 parent a06fe9f commit ce5bf57

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

trlx/trainer/accelerate_base_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ def evaluate(self): # noqa: C901
418418
rich_table.add_row(*[str(significant(x)) for x in rows[ix]])
419419

420420
if not ray.is_initialized():
421-
if "wandb" in self.config.train.trackers:
421+
if self.config.train.tracker == "wandb":
422422
import wandb
423423

424424
stats["samples"] = wandb.Table(columns, rows)

0 commit comments

Comments
 (0)