Method optimize_node

src/agents/optimization/node_optimizer.py:175–199

Optimizes the configuration of each node in the solution based on the cases. Args: case_list (list[Case]): The list of cases. solution (Solution): The solution to be optimized. Returns: dict: A dictionary with the optimization status and

(self, case_list: list[Case], solution: Solution)

            case.dump(os.path.join(save_dir, f"{case.case_id}.json"))

    def optimize_node(self, case_list: list[Case], solution: Solution):
        """
        Optimizes the configuration of each node in the solution based on the cases.

        Args:
            case_list (list[Case]): The list of cases.
            solution (Solution): The solution to be optimized.

        Returns:
            dict: A dictionary with the optimization status and method for each node.
        """
        node_name_list = []
        for state in case_list[0].trajectory.states:
            # Get unique node names based on case0's information and node_name is not duplicated
            if state.node.node_name not in node_name_list:
                node_name_list.append(state.node.node_name)

        # Do optimization for each node and get the optimization method
        op_info = {}
        for node_name in node_name_list:
            new_node, op_status, op_method = self.optimize_single_node(case_list, solution.sop.nodes[node_name])
            solution.sop.nodes[node_name] = new_node
            op_info[node_name] = {"optim_status": op_status, "optim_method": op_method}

        return op_info

    def optimize_single_node(self, case_list: list[Case], node: Node):
        """