Skip to content

Commit c6d8c62

Browse files
committed
Stripped useless argument from instrumentation class
1 parent 592ac51 commit c6d8c62

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

aalpy/learning_algs/general_passive/GeneralizedStateMerging.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def log_promote(self, node: GsmNode):
3535
def log_merge(self, part: Partitioning):
3636
pass
3737

38-
def learning_done(self, root: GsmNode, red_states: List[GsmNode]):
38+
def learning_done(self, root: GsmNode):
3939
pass
4040

4141

@@ -192,7 +192,7 @@ def run(self, data, convert=True, instrumentation: Instrumentation=None, data_fo
192192
# FUTURE: caching for aggregating compatibility tests
193193
partition_candidates.clear()
194194

195-
instrumentation.learning_done(root, red_states)
195+
instrumentation.learning_done(root)
196196

197197
root = self.postprocessing(root)
198198
if convert:

aalpy/learning_algs/general_passive/Instrumentation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,9 +60,9 @@ def log_merge(self, part: Partitioning):
6060
self.nr_merged_states += 1
6161
self.print_status()
6262

63-
def learning_done(self, root, red_states):
63+
def learning_done(self, root: GsmNode):
6464
print(f'\nLearning Time: {round(time.time() - self.previous_time, 2)}')
65-
print(f'Learned {len(red_states)} state automaton via {self.nr_merged_states} merges.')
65+
print(f'Learned {self.nr_red_states} state automaton via {self.nr_merged_states} merges.')
6666
if 2 < self.lvl:
6767
root.visualize("model", self.gsm.output_behavior)
6868

0 commit comments

Comments
 (0)