EventSequenceModel: Write error log message on TypeError (CART)

1 files changed, 10 insertions, 1 deletions

diff --git a/lib/behaviour.py b/lib/behaviour.py
index 402ddc7..1e59d20 100644
--- a/lib/behaviour.py
+++ b/lib/behaviour.py
@@ -66,7 +66,16 @@ class EventSequenceModel:
                     )
                 )
             except KeyError:
-                logging.error(f"Cannot predict {name}.{action}({param}) from LUT model")
+                if use_lut:
+                    logging.error(
+                        f"Cannot predict {name}.{action}({param}) from LUT model"
+                    )
+                else:
+                    logging.error(f"Cannot predict {name}.{action}({param}) from model")
+                raise
+            except TypeError:
+                if not use_lut:
+                    logging.error(f"Cannot predict {name}.{action}({param}) from model")
                 raise
 
             if aggregate == "sum":