licenses(["notice"])  # Apache 2.0

py_library(
    name = "gaussian_average_query",
    srcs = ["gaussian_average_query.py"],
    deps = [
        ":private_queries",
        "@org_tensorflow//tensorflow:tensorflow_py",
    ],
)

py_library(
    name = "dp_optimizers",
    deps = [
        ":dp_adam",
        ":dp_gradient_descent",
    ],
)

py_library(
    name = "dp_adam",
    srcs = [
        "dp_adam.py",
    ],
    deps = [
        ":gaussian_average_query",
        "@org_tensorflow//tensorflow:tensorflow_py",
    ],
)

py_library(
    name = "dp_gradient_descent",
    srcs = [
        "dp_gradient_descent.py",
    ],
    deps = [
        ":gaussian_average_query",
        "@org_tensorflow//tensorflow:tensorflow_py",
    ],
)

py_test(
    name = "dp_optimizer_test",
    srcs = ["dp_optimizer_test.py"],
    deps = [
        ":dp_optimizers",
        "@absl_py//absl/testing:parameterized",
        "@org_tensorflow//tensorflow:tensorflow_py",
    ],
)

py_library(
    name = "private_queries",
    srcs = ["private_queries.py"],
)