From 76832a396075c30cea7d272cff69dd75354a057c Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Tue, 16 Nov 2021 08:54:49 +0100 Subject: Add env var for safe functions; disable unsafe functions in that case --- lib/functions.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'lib') diff --git a/lib/functions.py b/lib/functions.py index 7cd06c0..0a488dc 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -740,19 +740,25 @@ class analytic: # ), } - if safe_functions_enabled: + if safe_functions_enabled or bool( + int(os.getenv("DFATOOL_REGRESSION_SAFE_FUNCTIONS", "0")) + ): + functions.pop("logarithmic1") + functions.pop("logarithmic") functions["safe_log"] = ParamFunction( lambda reg_param, model_param: reg_param[0] + reg_param[1] * analytic._safe_log(model_param), lambda model_param: True, 2, ) + functions.pop("inverse") functions["safe_inv"] = ParamFunction( lambda reg_param, model_param: reg_param[0] + reg_param[1] * analytic._safe_inv(model_param), lambda model_param: True, 2, ) + functions.pop("sqrt") functions["safe_sqrt"] = ParamFunction( lambda reg_param, model_param: reg_param[0] + reg_param[1] * analytic._safe_sqrt(model_param), -- cgit v1.2.3