add sub-state generation

1 files changed, 6 insertions, 1 deletions

diff --git a/lib/model.py b/lib/model.py
index d7a9bc9..57507e7 100644
--- a/lib/model.py
+++ b/lib/model.py
@@ -944,7 +944,12 @@ class PTAModel:
                     penalty = self.pelt.get_penalty_value(
                         self.by_param[k]["power_traces"]
                     )
-                    print(f"    penalty: {penalty}")
+                    print(
+                        f"    we found {penalty[1]} changepoints with penalty {penalty[0]}"
+                    )
+                    self.pelt.calc_raw_states(
+                        self.by_param[k]["power_traces"], penalty[0]
+                    )
 
         return None, None