From d3937efc54c3740ec9d7bed4b0319971bed9f742 Mon Sep 17 00:00:00 2001 From: kta-intel Date: Fri, 15 Nov 2024 13:41:36 -0800 Subject: [PATCH] revert space removal 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()