From e008e4ad9671dc641206a1dc797df497f8295f2b Mon Sep 17 00:00:00 2001 From: kta-intel Date: Fri, 15 Nov 2024 13:53:55 -0800 Subject: [PATCH] revert changes on interface.plan Signed-off-by: kta-intel --- openfl/interface/plan.py | 1 + 1 file changed, 1 insertion(+) diff --git a/openfl/interface/plan.py b/openfl/interface/plan.py index d235ad3070..a5e2fd9b2f 100644 --- a/openfl/interface/plan.py +++ b/openfl/interface/plan.py @@ -172,6 +172,7 @@ def initialize( ) data_loader = get_dataloader(plan, prefer_minimal=True, input_shape=input_shape) + task_runner = plan.get_task_runner(data_loader) tensor_pipe = plan.get_tensor_pipe()