diff --git a/examples/opacus/opacus-fl/server_app.py b/examples/opacus/opacus-fl/server_app.py index 7d3131e38a00..57c7d873091f 100644 --- a/examples/opacus/opacus-fl/server_app.py +++ b/examples/opacus/opacus-fl/server_app.py @@ -5,7 +5,7 @@ from flwr.common import Metrics, Context, ndarrays_to_parameters from flwr.server import ServerApp, ServerConfig, ServerAppComponents from flwr.server.strategy import FedAvg -from .task import Net, get_weights +from opacus_fl.task import Net, get_weights def weighted_average(metrics: List[Tuple[int, Metrics]]) -> Metrics: