set_node_supply -> set_node_supplies in python min cost flow
This commit is contained in:
@@ -32,7 +32,7 @@ PYBIND11_MODULE(min_cost_flow, m) {
|
||||
pybind11::vectorize(&SimpleMinCostFlow::AddArcWithCapacityAndUnitCost));
|
||||
smcf.def("set_node_supply", &SimpleMinCostFlow::SetNodeSupply, arg("node"),
|
||||
arg("supply"));
|
||||
smcf.def("set_nodes_supply",
|
||||
smcf.def("set_nodes_supplies",
|
||||
pybind11::vectorize(&SimpleMinCostFlow::SetNodeSupply));
|
||||
smcf.def("num_nodes", &SimpleMinCostFlow::NumNodes);
|
||||
smcf.def("num_arcs", &SimpleMinCostFlow::NumArcs);
|
||||
|
||||
@@ -70,7 +70,6 @@ def main():
|
||||
|
||||
# [START print_solution]
|
||||
if status == smcf.OPTIMAL:
|
||||
smcf.solve()
|
||||
print('Total cost = ', smcf.optimal_cost())
|
||||
print()
|
||||
for arc in range(smcf.num_arcs()):
|
||||
|
||||
@@ -47,7 +47,7 @@ def main():
|
||||
start_nodes, end_nodes, capacities, unit_costs)
|
||||
|
||||
# Add supply for each nodes.
|
||||
smcf.set_nodes_supply(np.arange(0, len(supplies)), supplies)
|
||||
smcf.set_nodes_supplies(np.arange(0, len(supplies)), supplies)
|
||||
# [END constraints]
|
||||
|
||||
# [START solve]
|
||||
|
||||
Reference in New Issue
Block a user