Fix tests

Signed-off-by: Gabriele A. Ron <[email protected]>
Macr0Nerd · Dec 29, 2023 · a6aeb06 · a6aeb06
1 parent 93d7b0f
commit a6aeb06
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/project_dilemma/object_loaders.py b/src/project_dilemma/object_loaders.py
@@ -23,7 +23,8 @@ def create_nodes(config: ProjectDilemmaConfig, algorithms_map: Dict[str, type[Al
 
     for node in config['nodes']:
         if not node.get('quantity'):
-            node['quantity'] = 1
+            nodes.append(Node(node["node_id"], algorithms_map[node['algorithm']['object']]))
+            continue
 
         for i in range(node['quantity']):
             nodes.append(Node(f'{node["node_id"]}{i}', algorithms_map[node['algorithm']['object']]))