analyze-trace: warn when encountering unexpected transition destination model

1 files changed, 7 insertions, 1 deletions

diff --git a/bin/analyze-trace.py b/bin/analyze-trace.py
index 9aa72d8..5da375f 100755
--- a/bin/analyze-trace.py
+++ b/bin/analyze-trace.py
@@ -216,7 +216,13 @@ def main():
                         i_to_transition[i] = t_to
                 am = AnalyticModel(am_tt_by_name, am_tt_param_names, force_tree=True)
                 model, info = am.get_fitted()
-                flat_model = info(name, t_from).flatten()
+                if type(info(name, t_from)) is df.SplitFunction:
+                    flat_model = info(name, t_from).flatten()
+                else:
+                    flat_model = list()
+                    logging.warning(
+                        f"Model for {name} {t_from} is {info(name, t_from)}, expected SplitFunction"
+                    )
 
                 for prefix, output in flat_model:
                     transition_name = i_to_transition[int(output)]