From a74d45c7f27306b39b764c3c0672b086fca5c00d Mon Sep 17 00:00:00 2001 From: kta-intel Date: Thu, 6 Jun 2024 15:09:22 -0700 Subject: [PATCH] cleaning up tf import lines Signed-off-by: kta-intel --- openfl/federated/task/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/openfl/federated/task/__init__.py b/openfl/federated/task/__init__.py index 084fc9506c..0838fc802d 100644 --- a/openfl/federated/task/__init__.py +++ b/openfl/federated/task/__init__.py @@ -16,8 +16,7 @@ from .runner import TaskRunner # NOQA if importlib.util.find_spec('tensorflow'): - from .runner_tf import TensorFlowTaskRunner # NOQA - from .runner_tf import TensorFlowTaskRunner_v1 # NOQA + from .runner_tf import TensorFlowTaskRunner, TensorFlowTaskRunner_v1 # NOQA from .runner_keras import KerasTaskRunner # NOQA from .fl_model import FederatedModel # NOQA if importlib.util.find_spec('torch'):