# buildifier: disable=same-origin-load
# Placeholder: load py_binary
# Placeholder: load py_library
load(
    "//lingvo:lingvo.bzl",
    "lingvo_py_binary",
)

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

licenses(["notice"])

py_binary(
    name = "download_lm1b",
    srcs = ["download_lm1b.py"],
    deps = [
        "//lingvo:compat",
    ],
)

py_library(
    name = "gshard_lm_decode_lib",
    srcs = ["gshard_lm_decode.py"],
    deps = [
        "//lingvo:compat",
        "//lingvo:model_registry",
        "//lingvo/core:gshard_decode",
        # Implicit numpy dependency.
    ],
)

lingvo_py_binary(
    name = "gshard_lm_decode",
    srcs = [":gshard_lm_decode_lib"],
    deps = [
        ":gshard_lm_decode_lib",
        "//lingvo:model_imports",
        # Implicit tensorflow grpc dependency.
    ],
)
