From c806d27c1d5d2495096be83911d84ea5ed43d0ee Mon Sep 17 00:00:00 2001
From: Birte Kristina Friesel <birte.friesel@uos.de>
Date: Fri, 9 Feb 2024 08:52:01 +0100
Subject: add --param-shift=param=categorial option

---
 lib/cli.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'lib')

diff --git a/lib/cli.py b/lib/cli.py
index e2c8779..fb1e722 100644
--- a/lib/cli.py
+++ b/lib/cli.py
@@ -583,7 +583,7 @@ def add_standard_arguments(parser):
     )
     parser.add_argument(
         "--param-shift",
-        metavar="<key>=<+|-|*|/><value>|none-to-0;...",
+        metavar="<key>=<+|-|*|/><value>|none-to-0|categorial;...",
         type=str,
         help="Adjust parameter values before passing them to model generation",
     )
@@ -681,6 +681,8 @@ def parse_shift_function(param_name, param_shift):
     elif param_shift.startswith("/"):
         param_shift_value = float(param_shift[1:])
         return lambda p: p / param_shift_value
+    elif param_shift == "categorial":
+        return lambda p: "=" + str(p)
     elif param_shift == "none-to-0":
         return lambda p: p or 0
     else:
-- 
cgit v1.2.3