minor bug fixed

Author: GioeleB00

src/app/core/runtime/edge.py

@@ -6,8 +6,8 @@
 waits the sampled delay (and any resource wait) before delivering the
 message to the target node's inbox.
 """
-
 from collections.abc import Generator
+from typing import TYPE_CHECKING
 
 import numpy as np
 import simpy
@@ -17,6 +17,10 @@
 from app.core.event_samplers.common_helpers import general_sampler
 from app.schemas.system_topology_schema.full_system_topology_schema import Edge
 
+if TYPE_CHECKING:
+    from app.schemas.random_variables_config import RVConfig
+
+
 
 class EdgeRuntime:
     """definining the logic to handle the edges during the simulation"""
@@ -37,16 +41,16 @@ def __init__(
 
     def _deliver(self, state: RequestState) -> Generator[simpy.Event, None, None]:
         """Function to deliver the state to the next node"""
-        # assign a probability that the network will drop the request
-        distribution = self.edge_config.latency.distribution
+        # extract the random variables defining the latency of the edge
+        random_variable: RVConfig = self.edge_config.latency
 
         uniform_variable = self.rng.uniform()
         if uniform_variable < NetworkParameters.DROPOUT_RATE:
             state.finish_time = self.env.now
             state.record_hop(f"{self.edge_config.id}-dropped")
             return
 
-        transit_time = general_sampler(distribution, self.rng)
+        transit_time = general_sampler(random_variable, self.rng)
         yield self.env.timeout(transit_time)
         state.record_hop(self.edge_config.id, self.env.now)
         self.target_box.put(state)

tests/unit/input_sructure/test_full_topology_input.py

@@ -16,8 +16,8 @@
 
 from app.config.constants import (
     EndpointStepCPU,
-    StepOperation,
     ServerResourcesDefaults,
+    StepOperation,
     SystemEdges,
     SystemNodes,
 )