linear_solver: backport from main

This commit is contained in:
Corentin Le Molgat
2025-10-16 13:06:55 +02:00
parent 87f3383057
commit 28cfca9a22
4 changed files with 305 additions and 145 deletions

View File

@@ -11,22 +11,26 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# Python wrapper for model_builder.
load("@pip_deps//:requirements.bzl", "requirement")
load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
load("@rules_python//python:py_binary.bzl", "py_binary")
load("@rules_python//python:py_library.bzl", "py_library")
load("@rules_python//python:py_test.bzl", "py_test")
# Python wrapper for model_builder.
pybind_extension(
name = "model_builder_helper",
srcs = ["model_builder_helper.cc"],
visibility = ["//visibility:public"],
deps = [
"//ortools/base:logging",
"//ortools/linear_solver:linear_solver_cc_proto",
"//ortools/linear_solver:model_exporter",
"//ortools/linear_solver/wrappers:model_builder_helper",
"@abseil-cpp//absl/hash",
"@abseil-cpp//absl/log:check",
"@abseil-cpp//absl/strings",
"@abseil-cpp//absl/types:span",
"@eigen",
"@pybind11_protobuf//pybind11_protobuf:native_proto_caster",
],
@@ -41,11 +45,10 @@ py_test(
],
deps = [
":model_builder_helper",
":model_builder_numbers",
"//ortools/linear_solver:linear_solver_py_pb2",
requirement("absl-py"),
requirement("numpy"),
requirement("scipy"),
"//ortools/linear_solver:linear_solver_py_pb2",
],
)
@@ -80,18 +83,22 @@ py_test(
],
deps = [
":model_builder",
":model_builder_helper",
"@protobuf//:protobuf_python",
requirement("absl-py"),
requirement("numpy"),
requirement("pandas"),
"//ortools/linear_solver:linear_solver_py_pb2",
],
)
## Proto solver
py_binary(
name = "solve_model",
srcs = ["solve_model.py"],
deps = [
":model_builder",
requirement("absl-py"),
"//ortools/linear_solver:linear_solver_py_pb2",
],
)