#2248 - Removed redundant Union from single type params

src/primaite/simulator/network/container.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional
 
 import matplotlib.pyplot as plt
 import networkx as nx
@@ -272,9 +272,7 @@ class Network(SimComponent):
         self._node_request_manager.remove_request(name=node.hostname)
         _LOGGER.info(f"Removed node {node.hostname} from network {self.uuid}")
 
-    def connect(
-        self, endpoint_a: Union[WiredNetworkInterface], endpoint_b: Union[WiredNetworkInterface], **kwargs
-    ) -> Optional[Link]:
+    def connect(self, endpoint_a: WiredNetworkInterface, endpoint_b: WiredNetworkInterface, **kwargs) -> Optional[Link]:
         """
         Connect two endpoints on the network by creating a link between their NICs/SwitchPorts.
 

src/primaite/simulator/network/hardware/base.py

@@ -509,9 +509,9 @@ class Link(SimComponent):
     :param bandwidth: The bandwidth of the Link in Mbps (default is 100 Mbps).
     """
 
-    endpoint_a: Union[WiredNetworkInterface]
+    endpoint_a: WiredNetworkInterface
     "The first WiredNetworkInterface connected to the Link."
-    endpoint_b: Union[WiredNetworkInterface]
+    endpoint_b: WiredNetworkInterface
     "The second WiredNetworkInterface connected to the Link."
     bandwidth: float = 100.0
     "The bandwidth of the Link in Mbps (default is 100 Mbps)."
@@ -596,7 +596,7 @@ class Link(SimComponent):
             return True
         return False
 
-    def transmit_frame(self, sender_nic: Union[WiredNetworkInterface], frame: Frame) -> bool:
+    def transmit_frame(self, sender_nic: WiredNetworkInterface, frame: Frame) -> bool:
         """
         Send a network frame from one NIC or SwitchPort to another connected NIC or SwitchPort.