From 40a612ba666f163748cde33009573832154b8117 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Fri, 19 Jul 2024 17:10:45 +0200 Subject: Add DFATOOL_ULS_FUNCTIONS variable --- lib/functions.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'lib') diff --git a/lib/functions.py b/lib/functions.py index 0a00f83..ae8283a 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -2197,6 +2197,12 @@ class analytic: if os.getenv("DFATOOL_RMT_SUBMODEL", "uls") == "fol": functions = {"linear": functions["linear"]} + elif allowed_functions := os.getenv("DFATOOL_ULS_FUNCTIONS", None): + allowed_functions = allowed_functions.split(",") + all_functions = list(functions.keys()) + for function_name in all_functions: + if function_name not in allowed_functions: + functions.pop(function_name) return functions -- cgit v1.2.3