Skip to content

GitLab

Commit eb611e2c authored by Hejing Li's avatar Hejing Li
Browse files

net/base.py: rename add function

parent a72c6391
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
experiments/simbricks/orchestration/simulation/net/base.py
View file @ eb611e2c
...@@ -78,9 +78,10 @@ class WireNet(NetSim): ...@@ -78,9 +78,10 @@ class WireNet(NetSim):
self._wire_comp: eth.EthWire | None = None self._wire_comp: eth.EthWire | None = None
self._relative_pcap_file_path: str | None = 'relative_pcap_file_path' self._relative_pcap_file_path: str | None = 'relative_pcap_file_path'
def add_wire(self, wire: eth.EthWire): def add(self, wire: eth.EthWire):
assert self._wire_comp is None assert self._wire_comp is None
super()._add_component(wire) super()._add_component(wire)
self.experiment.add_network(self)
self._wire_comp = wire self._wire_comp = wire
def run_cmd(self, inst: inst_base.Instantiation) -> str: def run_cmd(self, inst: inst_base.Instantiation) -> str:
...@@ -130,9 +131,10 @@ class SwitchNet(NetSim): ...@@ -130,9 +131,10 @@ class SwitchNet(NetSim):
self._switch_spec: eth.EthSwitch | None = None self._switch_spec: eth.EthSwitch | None = None
self._relative_pcap_file_path: str | None = relative_pcap_file_path self._relative_pcap_file_path: str | None = relative_pcap_file_path
def add_switch(self, switch_spec: eth.EthSwitch): def add(self, switch_spec: eth.EthSwitch):
assert self._switch_spec is None assert self._switch_spec is None
super()._add_component(switch_spec) super()._add_component(switch_spec)
self.experiment.add_network(self)
self._switch_spec = switch_spec self._switch_spec = switch_spec
def run_cmd(self, inst: inst_base.Instantiation) -> str: def run_cmd(self, inst: inst_base.Instantiation) -> str:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment