Merge branch 'replace-fwd-bwd-flowercontext' of https://github.com/ad…

…ap/flower into replace-fwd-bwd-flowercontext

src/py/flwr/client/middleware/utils_test.py

@@ -44,7 +44,7 @@ def make_mock_middleware(name: str, footprint: List[str]) -> Layer:
 
     def middleware(message: Message, context: Context, app: FlowerCallable) -> Message:
         footprint.append(name)
-        # add empty ConfigRegcord to in_message for this middleware layer
+        # add empty ConfigRecord to in_message for this middleware layer
         message.message.set_configs(name=name, record=ConfigsRecord())
         _increment_context_counter(context)
         out_message: Message = app(message, context)

src/py/flwr/client/typing.py

@@ -14,32 +14,13 @@
 # ==============================================================================
 """Custom types for Flower clients."""
 
-from dataclasses import dataclass
 from typing import Callable
 
 from flwr.common.context import Context
 from flwr.common.message import Message
-from flwr.proto.task_pb2 import TaskIns, TaskRes  # pylint: disable=E0611
 
 from .client import Client as Client
 
-
-@dataclass
-class Fwd:
-    """."""
-
-    task_ins: TaskIns
-    context: Context
-
-
-@dataclass
-class Bwd:
-    """."""
-
-    task_res: TaskRes
-    context: Context
-
-
 FlowerCallable = Callable[[Message, Context], Message]
 ClientFn = Callable[[str], Client]
 Layer = Callable[[Message, Context, FlowerCallable], Message]

src/py/flwr/flower/__init__.py

@@ -16,11 +16,7 @@
 
 
 from flwr.client.flower import Flower as Flower
-from flwr.client.typing import Bwd as Bwd
-from flwr.client.typing import Fwd as Fwd
 
 __all__ = [
     "Flower",
-    "Fwd",
-    "Bwd",
 ]