set_node_supply -> set_node_supplies in python min cost flow

This commit is contained in:
Laurent Perron
2022-12-07 13:22:53 +01:00
parent 2187b58ae9
commit 92ebf24647
3 changed files with 2 additions and 3 deletions

View File

@@ -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);

View File

@@ -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()):

View File

@@ -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]