FOL: Improve error messages

1 files changed, 6 insertions, 1 deletions

diff --git a/lib/functions.py b/lib/functions.py
index 7cc8cdd..9a99c58 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -719,9 +719,14 @@ class FOLFunction(ModelFunction):
             return self._function(self.model_args, actual_param_list)
         except FloatingPointError as e:
             logger.error(
-                f"{e} when predicting {self._function_str}({param_list}), returning static value"
+                f"{e} when predicting {self._function_str}({self.model_args}, {actual_param_list}) for {param_list}, returning static value"
             )
             return self.value
+        except TypeError as e:
+            logger.error(
+                f"{e} when predicting {self._function_str}({self.model_args}, {actual_param_list}) for {param_list}"
+            )
+            raise
 
     def to_json(self, **kwargs):
         ret = super().to_json(**kwargs)