Skip to content

Commit 8181580

Browse files
committed
Updates Node model and cleanup
1 parent 6615e9b commit 8181580

File tree

3 files changed

+9
-1
lines changed
  • packages/models-library/src/models_library
  • services/director-v2/src/simcore_service_director_v2

3 files changed

+9
-1
lines changed

packages/models-library/src/models_library/projects_nodes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -383,7 +383,8 @@ def _update_json_schema_extra(schema: JsonDict) -> None:
383383

384384
model_config = ConfigDict(
385385
extra="forbid",
386-
populate_by_name=True,
386+
validate_by_name=True,
387+
validate_by_alias=True,
387388
json_schema_extra=_update_json_schema_extra,
388389
)
389390

services/director-v2/src/simcore_service_director_v2/modules/db/repositories/comp_tasks/_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -400,6 +400,9 @@ async def generate_tasks_list_from_project(
400400
raise WalletNotEnoughCreditsError(
401401
wallet_name=wallet_info.wallet_name,
402402
wallet_credit_amount=wallet_info.wallet_credit_amount,
403+
user_id=user_id,
404+
product_name=product_name,
405+
project_id=project.uuid,
403406
)
404407

405408
assert rabbitmq_rpc_client # nosec

services/director-v2/src/simcore_service_director_v2/utils/dags.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ def create_complete_dag(workbench: NodesDict) -> nx.DiGraph:
2929
dag_graph: nx.DiGraph = nx.DiGraph()
3030
for node_id, node in workbench.items():
3131
assert node.state # nosec
32+
3233
dag_graph.add_node(
3334
node_id,
3435
name=node.label,
@@ -43,6 +44,9 @@ def create_complete_dag(workbench: NodesDict) -> nx.DiGraph:
4344
if node.input_nodes:
4445
for input_node_id in node.input_nodes:
4546
predecessor_node = workbench.get(NodeIDStr(input_node_id))
47+
assert (
48+
predecessor_node
49+
), f"Node {input_node_id} not found in workbench" # nosec
4650
if predecessor_node:
4751
dag_graph.add_edge(str(input_node_id), node_id)
4852

0 commit comments

Comments
 (0)