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