diff --git a/environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/engine/run.py b/environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/engine/run.py index 03f1695c..4c1baa45 100644 --- a/environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/engine/run.py +++ b/environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/engine/run.py @@ -152,7 +152,10 @@ class Run: f.write(str(train_time) + "\n") def _validate( - self, validator: Trainer, model: LightningModule, data_module: LightningDataModule + self, + validator: Trainer, + model: LightningModule, + data_module: LightningDataModule, ) -> _EVALUATE_OUTPUT: score = validator.validate(model, datamodule=data_module) return score