# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

load("//:rules.bzl", "runai_cc_auto_library", "runai_cc_test")

runai_cc_auto_library(
    name = "batches",
    deps = [
        "//common/device",
        "//streamer/impl/config/config",
        "//streamer/impl/batch",
        "//common/s3_wrapper",
        "//common/exception",
        "//common/responder",
        "//streamer/impl/file",
        "//streamer/impl/assigner/file_read_task",
        "//streamer/impl/s3",
        "//utils/logging",
    ],
    linkopts = [ "-lm" ]
)

runai_cc_test(
    name = "batches_test",
    srcs = ["batches_test.cc"],
    deps = [
        ":batches",
        "//utils/random",
        "//utils/temp/env",
        "//utils/temp/file",
        "//streamer/impl/assigner",
    ],
)
