oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2248 - Removed redundant Union from single type params

Browse Source
This commit is contained in:
Chris McCarthy
2024-02-09 11:37:47 +00:00
parent cceb6208e0
commit 6b3829dc48
2 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/primaite/simulator/network/container.py
View File

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

6
src/primaite/simulator/network/hardware/base.py
View File

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