cleaning up tf import lines

Signed-off-by: kta-intel <[email protected]>
securefederatedai · Jun 6, 2024 · a74d45c · a74d45c
1 parent 2018e1f
commit a74d45c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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'):