BUILD

# buildifier: disable=out-of-order-load - Breaks copybara otherwise
load("//third_party/bazel_rules/rules_python/python:py_binary.bzl", "py_binary")
load("//sonnet/src:build_defs.bzl", "snt_py_library", "snt_py_test")

package(default_visibility = ["//visibility:private"])

licenses(["notice"])

py_binary(
    name = "simple_mnist",
    srcs = ["simple_mnist.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        # pip: absl:app
        "//sonnet",
        # pip: tensorflow
        # pip: tensorflow_datasets
    ],
)

snt_py_library(
    name = "simple_mnist_library",
    srcs = ["simple_mnist.py"],
    deps = [
        # pip: absl:app
        "//sonnet",
        # pip: tensorflow
        # pip: tensorflow_datasets
    ],
)

snt_py_test(
    name = "simple_mnist_test",
    srcs = ["simple_mnist_test.py"],
    deps = [
        ":simple_mnist_library",
        "//sonnet",
        "//sonnet/src:test_utils",
        # pip: tensorflow
    ],
)

py_binary(
    name = "functional_mlp_mnist",
    srcs = ["functional_mlp_mnist.py"],
    python_version = "PY3",
    srcs_version = "PY3",
    deps = [
        # pip: absl:app
        # pip: absl/logging
        "//sonnet",
        # pip: tensorflow
        # pip: tensorflow_datasets
    ],
)