Fix circular import using absolute imports

Signed-off-by: Shah, Karan <[email protected]>
securefederatedai · Jun 6, 2024 · cafb965 · cafb965
1 parent 555d8e3
commit cafb965
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git 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
@@ -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
@@ -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