From 3ca88bdc021366d4b54e7f638c8bd446ed42f500 Mon Sep 17 00:00:00 2001 From: Ynon Flum Date: Mon, 16 Dec 2024 13:10:59 +0200 Subject: [PATCH] Fix lint error --- openfl/federated/task/fl_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openfl/federated/task/fl_model.py b/openfl/federated/task/fl_model.py index eb7365225f9..8029cc08196 100644 --- a/openfl/federated/task/fl_model.py +++ b/openfl/federated/task/fl_model.py @@ -77,7 +77,7 @@ def __init__(self, build_model, optimizer=None, loss_fn=None, **kwargs): self.runner.validate = lambda *args, **kwargs: build_model.validate( self.runner, *args, **kwargs ) - + if hasattr(self.model, "train_epoch"): self.runner.train_epoch = lambda *args, **kwargs: build_model.train_epoch( self.runner, *args, **kwargs