Add DFATOOL_ULS_FUNCTIONS variable

1 files changed, 6 insertions, 0 deletions

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