🐛 fix nargs + or * (#140)

Signed-off-by: Joe Runde <[email protected]>
opendatahub-io · Sep 24, 2024 · 9031668 · 9031668
1 parent 5852586
commit 9031668
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/vllm_tgis_adapter/tgis_utils/args.py b/src/vllm_tgis_adapter/tgis_utils/args.py
@@ -39,7 +39,11 @@ def _switch_action_default(action: argparse.Action) -> None:
         # for non-string args, the string value of the env var will be parsed
         # based on the action.type when setting from the default value
         val = env_val
-    action.default = val
+
+    if action.nargs in ("+", "*"):
+        action.default = [val]
+    else:
+        action.default = val
 
 
 class EnvVarArgumentParser(FlexibleArgumentParser):