fix(AutoML): fixed multiple architecture passing for advanced training

frankkramer-lab · Jun 30, 2022 · f4b165c · f4b165c
1 parent c31355d
commit f4b165c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/aucmedi/automl/parser_cli.py b/aucmedi/automl/parser_cli.py
@@ -35,6 +35,9 @@ def parse_cli(args):
     if config["hub"] == "training":
         # Handle 3D shape - from str to tuple
         config["shape_3D"] = tuple(map(int, config["shape_3D"].split("x")))
+        # Handle architecture list
+        if "," in config["architecture"]:
+            config["architecture"] = config["architecture"].split(",")
     else:
         # Handle input path
         config["input"] = config["path_input"]