mypy

Author: pcrespov

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

@@ -78,20 +78,23 @@ class NodeState(BaseModel):
             description="true if the node's outputs need to be re-computed",
         ),
     ] = True
+
     dependencies: Annotated[
         set[NodeID],
         Field(
             default_factory=set,
             description="contains the node inputs dependencies if they need to be computed first",
         ),
     ] = DEFAULT_FACTORY
+
     current_status: Annotated[
         RunningState,
         Field(
             description="the node's current state",
             alias="currentStatus",
         ),
     ] = RunningState.NOT_STARTED
+
     progress: Annotated[
         float | None,
         Field(
@@ -103,6 +106,7 @@ class NodeState(BaseModel):
 
     model_config = ConfigDict(
         extra="forbid",
+        populate_by_name=True,
         json_schema_extra={
             "examples": [
                 {
@@ -284,7 +288,7 @@ def _convert_old_enum_name(v) -> RunningState:
 
             # the old version of state was a enum of RunningState
             running_state_value = _convert_old_enum_name(v)
-            return NodeState.model_validate({"currentStatus": running_state_value})
+            return NodeState(current_status=running_state_value)
         return v
 
     @staticmethod

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

@@ -246,7 +246,7 @@ async def compute_pipeline_details(
             node_id: NodeState(
                 modified=node_data.get(kNODE_MODIFIED_STATE, False),
                 dependencies=node_data.get(kNODE_DEPENDENCIES_TO_COMPUTE, set()),
-                currentStatus=node_id_to_comp_task[node_id].state,
+                current_status=node_id_to_comp_task[node_id].state,
                 progress=(
                     node_id_to_comp_task[node_id].progress
                     if node_id_to_comp_task[node_id].progress is not None

services/web/server/src/simcore_service_webserver/projects/_projects_service.py

@@ -1024,7 +1024,7 @@ async def update_project_node_state(
         project_id=project_id,
         node_id=node_id,
         partial_node=PartialNode.model_construct(
-            state=NodeState(currentStatus=RunningState(new_state))  # type: ignore[call-arg]
+            state=NodeState(current_status=RunningState(new_state))
         ),
     )