From d9930527a42dafb050dea8ef2bc86f8f83f8a0ec Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Fri, 10 Jun 2022 15:32:50 +0200
Subject: param-shift: add none-to-0 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 6b5b796..e0cbc7f 100644
--- a/lib/cli.py
+++ b/lib/cli.py
@@ -255,7 +255,7 @@ def add_standard_arguments(parser):
     )
     parser.add_argument(
         "--param-shift",
-        metavar="<key>=<+|-|*|/><value>;...",
+        metavar="<key>=<+|-|*|/><value>|none-to-0;...",
         type=str,
         help="Adjust parameter values before passing them to model generation",
     )
@@ -285,6 +285,8 @@ def parse_param_shift(raw_param_shift):
         elif param_shift.startswith("/"):
             param_shift_value = float(param_shift[1:])
             param_shift_function = lambda p: p / param_shift_value
+        elif param_shift == "none-to-0":
+            param_shift_function = lambda p: p or 0
         else:
             raise ValueError(f"Unsupported shift operation {param_name}={param_shift}")
         shift_list.append((param_name, param_shift_function))
-- 
cgit v1.2.3