xref: /aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/python/BUILD (revision 14675a029014e728ec732f129a32e299b2da0601) 
1load("@org_tensorflow//tensorflow:tensorflow.bzl", "pybind_extension")
2load("@rules_python//python:defs.bzl", "py_test")
3
4package(
5    default_visibility = ["//fcp/aggregation:internal"],
6)
7
8pybind_extension(
9    name = "aggregation_protocols",
10    srcs = ["aggregation_protocols.cc"],
11    pytype_deps = [
12        "//fcp/aggregation/protocol/python:aggregation_protocol",
13    ],
14    deps = [
15        "//fcp/aggregation/protocol:aggregation_protocol",
16        "//fcp/aggregation/protocol:configuration_cc_proto",
17        "//fcp/aggregation/protocol:resource_resolver",
18        "//fcp/aggregation/protocol/simple_aggregation",
19        "//fcp/aggregation/tensorflow:tensorflow_checkpoint_builder_factory",
20        "//fcp/aggregation/tensorflow:tensorflow_checkpoint_parser_factory",
21        "@com_google_absl//absl/status:statusor",
22        "@pybind11_abseil//pybind11_abseil:status_casters",
23        "@pybind11_protobuf//pybind11_protobuf:native_proto_caster",
24    ],
25)
26
27py_test(
28    name = "aggregation_protocols_test",
29    srcs = ["aggregation_protocols_test.py"],
30    data = ["@pybind11_abseil//pybind11_abseil:status.so"],
31    deps = [
32        ":aggregation_protocols",
33        "//fcp/aggregation/protocol:configuration_py_pb2",
34        "//fcp/aggregation/protocol:py_pb2",
35    ],
36)
37