Fix trainer init

This commit is contained in:
mbsantiago 2025-06-26 19:40:27 -06:00
parent ce15a0f152
commit b396d4908a

View File

@ -101,8 +101,7 @@ def build_trainer(
targets: TargetProtocol,
) -> Trainer:
trainer_conf = PLTrainerConfig.model_validate(
conf.train,
from_attributes=True,
conf.train.model_dump(mode="python")
)
return Trainer(
**trainer_conf.model_dump(exclude_none=True),