From cafb965daa3136ccc43689d157fdc2dc0980ff2d Mon Sep 17 00:00:00 2001 From: "Shah, Karan" Date: Thu, 6 Jun 2024 13:17:38 +0530 Subject: [PATCH] Fix circular import using absolute imports Signed-off-by: Shah, Karan --- openfl/experimental/runtime/__init__.py | 8 +++----- openfl/experimental/runtime/federated_runtime.py | 2 +- openfl/experimental/runtime/local_runtime.py | 2 +- 3 files changed, 5 insertions(+), 7 deletions(-) diff --git a/openfl/experimental/runtime/__init__.py b/openfl/experimental/runtime/__init__.py index 608e5f5fb6..195b42fe5d 100644 --- a/openfl/experimental/runtime/__init__.py +++ b/openfl/experimental/runtime/__init__.py @@ -2,8 +2,6 @@ # SPDX-License-Identifier: Apache-2.0 """ openfl.experimental.runtime package Runtime class.""" -from .federated_runtime import FederatedRuntime -from .local_runtime import LocalRuntime -from .runtime import Runtime - -__all__ = ["FederatedRuntime", "LocalRuntime", "Runtime"] +from openfl.experimental.runtime.federated_runtime import FederatedRuntime +from openfl.experimental.runtime.local_runtime import LocalRuntime +from openfl.experimental.runtime.runtime import Runtime diff --git a/openfl/experimental/runtime/federated_runtime.py b/openfl/experimental/runtime/federated_runtime.py index 056218ff18..bae51c8fe3 100644 --- a/openfl/experimental/runtime/federated_runtime.py +++ b/openfl/experimental/runtime/federated_runtime.py @@ -6,7 +6,7 @@ from typing import TYPE_CHECKING -from openfl.experimental.runtime import Runtime +from openfl.experimental.runtime.runtime import Runtime if TYPE_CHECKING: from openfl.experimental.interface import Aggregator diff --git a/openfl/experimental/runtime/local_runtime.py b/openfl/experimental/runtime/local_runtime.py index 8a7881c474..19f8de0041 100644 --- a/openfl/experimental/runtime/local_runtime.py +++ b/openfl/experimental/runtime/local_runtime.py @@ -13,7 +13,7 @@ import ray -from openfl.experimental.runtime import Runtime +from openfl.experimental.runtime.runtime import Runtime if TYPE_CHECKING: from openfl.experimental.interface import Aggregator, Collaborator, FLSpec