Class Graph

src/utils/max_flow.py:6–16

class Graph:
    def __init__(self, node_count: int, edges: Dict[Tuple[int, int], int]):
        """
        edges: {(source, target): edge_weight}}
        """
        self.node_count = node_count
        self.edges = edges

    def iterate_edges(self) -> Iterable[Tuple[int, int, int]]:
        for (source, target), weight in self.edges.items():
            yield source, target, weight


class Edge(BaseModel):