small changes

Author: GioeleB00

src/asyncflow/runtime/actors/routing/lb_algorithms.py

@@ -1,11 +1,12 @@
 """algorithms to simulate the load balancer during the simulation"""
 import random
 from collections import OrderedDict
-from collections.abc import Callable, Mapping
+from collections.abc import Callable
 
 from asyncflow.config.enums import LbAlgorithmsName
 from asyncflow.runtime.actors.edge import EdgeRuntime
 
+
 def least_connections(
     edges: OrderedDict[str, EdgeRuntime],
     ) -> EdgeRuntime:

src/asyncflow/runtime/actors/server.py

@@ -104,7 +104,7 @@ def __init__( # noqa: PLR0913
         self._server_enabled_metrics = build_server_metrics(
             settings.enabled_sample_metrics,
         )
-        
+
         # Per-request metrics are keyed by request_id (int), not by RequestState object:
         # - ints are stable, lightweight, and hash/GC-friendly
         # - avoids holding strong refs to RequestState (no memory leaks)

tests/unit/runtime/actors/test_lb_algo.py

@@ -15,7 +15,6 @@
 from asyncflow.config.enums import LbAlgorithmsName
 from asyncflow.runtime.actors.routing.lb_algorithms import (
     LB_TABLE,
-    fcfs_picker,
     least_connections,
     random_choice,
     round_robin,
@@ -42,36 +41,6 @@ def _mk_edges(pairs: list[tuple[str, int]]) -> OrderedDict[str, _DummyEdge]:
     return OrderedDict((k, _DummyEdge(cc)) for k, cc in pairs)
 
 
-# ------------------------------- FCFS picker -------------------------------- #
-
-def test_fcfs_picker_returns_first_free_without_mutation() -> None:
-    """FCFS picker must return first *free* edge and not mutate the dict."""
-    od = _mk_edges([("a", 2), ("b", 1), ("c", 3)])
-    edges = cast("OrderedDict[str, EdgeRuntime]", od)
-
-    # a busy, b free, c busy
-    busy = {"a": 1, "b": 0, "c": 2}
-
-    pick = fcfs_picker(edges, busy)
-    assert pick is not None
-    edge_id, edge_rt = pick
-    assert edge_id == "b"
-    assert cast("_DummyEdge", edge_rt) is od["b"]
-
-    # Ensure no mutation of the original OrderedDict order/size
-    assert list(od.keys()) == ["a", "b", "c"]
-
-
-def test_fcfs_picker_all_busy_returns_none() -> None:
-    """FCFS picker should return None if no edges are free."""
-    od = _mk_edges([("a", 0), ("b", 0)])
-    edges = cast("OrderedDict[str, EdgeRuntime]", od)
-
-    busy = {"a": 1, "b": 3}
-    pick = fcfs_picker(edges, busy)
-    assert pick is None
-
-
 # ----------------------------- Other algorithms ----------------------------- #
 
 def test_round_robin_rotates_and_returns_first() -> None: