Skip to content

Commit c10f4e0

Browse files
fix(cli): handle report_to as string or list
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 400fd3d commit c10f4e0

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/darija_translator/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
def build_trainer(model, tokenizer, train_dataset, eval_dataset,
1010
config: TrainConfig):
11-
if config.report_to == "wandb":
11+
if config.report_to == "wandb" or (isinstance(config.report_to, list) and "wandb" in config.report_to):
1212
os.environ["WANDB_PROJECT"] = config.wandb_project
1313

1414
trainer = SFTTrainer(

0 commit comments

Comments
 (0)