85 lines
1.1 KiB
Python
85 lines
1.1 KiB
Python
load("@rules_python//python:defs.bzl", "py_library", "py_test")
|
|
|
|
package(default_visibility=["//visibility:public"])
|
|
|
|
py_library(
|
|
name="config",
|
|
srcs=["config.py"],
|
|
deps=[
|
|
":logger",
|
|
":constants",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="constants",
|
|
srcs=["constants.py"],
|
|
)
|
|
|
|
|
|
py_library(
|
|
name = "envvar",
|
|
srcs = ["envvar.py"],
|
|
)
|
|
|
|
py_test(
|
|
name = "envvar_test",
|
|
srcs = ["envvar_test.py"],
|
|
deps = [":envvar"],
|
|
)
|
|
|
|
py_library(
|
|
name = "images",
|
|
srcs = ["images.py"],
|
|
)
|
|
|
|
py_library(
|
|
name="logger",
|
|
srcs=["logger.py"],
|
|
)
|
|
|
|
py_test(
|
|
name="logger_test",
|
|
srcs=["logger_test.py"],
|
|
deps=[":logger"],
|
|
)
|
|
|
|
py_library(
|
|
name="memory",
|
|
srcs=["memory.py"],
|
|
)
|
|
|
|
py_library(
|
|
name="model",
|
|
srcs=["model.py"],
|
|
deps=[
|
|
":constants",
|
|
":memory",
|
|
":logger",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="text2img",
|
|
srcs=["text2img.py"],
|
|
deps=[
|
|
":config",
|
|
":logger",
|
|
":images",
|
|
":memory",
|
|
":model",
|
|
":times",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="times",
|
|
srcs=["times.py"],
|
|
)
|
|
|
|
py_test(
|
|
name="times_test",
|
|
srcs=["times_test.py"],
|
|
deps=[":times"],
|
|
)
|