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

eod commit

Browse Source
This commit is contained in:
Marek Wolan
2024-02-23 10:06:48 +00:00
parent 72f4cc0a50
commit f933341df5
6 changed files with 24 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -45,12 +45,12 @@ class Network(SimComponent):
self._nx_graph = MultiGraph()
def reset_component_for_episode(self, episode: int):
def setup_for_episode(self, episode: int):
"""Reset the original state of the SimComponent."""
for node in self.nodes.values():
node.reset_component_for_episode(episode)
node.setup_for_episode(episode)
for link in self.links.values():
link.reset_component_for_episode(episode)
link.setup_for_episode(episode)
for node in self.nodes.values():
node.power_on()
@@ -171,7 +171,7 @@ class Network(SimComponent):
def clear_links(self):
"""Clear all the links in the network by resetting their component state for the episode."""
for link in self.links.values():
link.reset_component_for_episode()
link.setup_for_episode()
def draw(self, seed: int = 123):
"""