Skip to content

Commit

Permalink
[lint] buildifier format dag BUILD file (ray-project#46217)
Browse files Browse the repository at this point in the history
pure formatting change

Signed-off-by: Lonnie Liu <[email protected]>
  • Loading branch information
aslonnie committed Jun 24, 2024
1 parent 4819eec commit 560d968
Showing 1 changed file with 79 additions and 33 deletions.
112 changes: 79 additions & 33 deletions python/ray/dag/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -2,21 +2,26 @@ load("//bazel:python.bzl", "doctest")
load("//bazel:python.bzl", "py_test_module_list")

doctest(
files = glob(["**/*.py"], exclude=["**/experimental/**/*.py"]),
files = glob(
["**/*.py"],
exclude = ["**/experimental/**/*.py"],
),
tags = ["team:core"],
deps = [":dag_lib"]
deps = [":dag_lib"],
)


# This is a dummy test dependency that causes the above tests to be
# re-run if any of these files changes.
py_library(
name = "dag_lib",
srcs = glob(["**/*.py"], exclude=["tests/**/*.py"]),
srcs = glob(
["**/*.py"],
exclude = ["tests/**/*.py"],
),
visibility = [
"//python/ray/dag:__pkg__",
"//python/ray/dag:__subpackages__",
"//release:__pkg__"
"//release:__pkg__",
],
)

Expand All @@ -26,77 +31,118 @@ py_test(
name = "test_function_dag",
size = "small",
srcs = dag_tests_srcs,
tags = ["exclusive", "team:core", "ray_dag_tests"],
tags = [
"exclusive",
"ray_dag_tests",
"team:core",
],
deps = [":dag_lib"],
)

py_test(
name = "test_class_dag",
size = "small",
srcs = dag_tests_srcs,
tags = ["exclusive", "team:core", "ray_dag_tests"],
tags = [
"exclusive",
"ray_dag_tests",
"team:core",
],
deps = [":dag_lib"],
)

py_test(
name = "test_input_node",
size = "small",
srcs = dag_tests_srcs,
tags = ["exclusive", "team:core", "ray_dag_tests"],
tags = [
"exclusive",
"ray_dag_tests",
"team:core",
],
deps = [":dag_lib"],
)

py_test(
name = "test_output_node",
size = "small",
srcs = dag_tests_srcs,
tags = ["exclusive", "team:core", "ray_dag_tests"],
tags = [
"exclusive",
"ray_dag_tests",
"team:core",
],
deps = [":dag_lib"],
)

py_test(
name = "test_plot",
size = "small",
srcs = dag_tests_srcs,
tags = ["exclusive", "team:core", "ray_dag_tests"],
tags = [
"exclusive",
"ray_dag_tests",
"team:core",
],
deps = [":dag_lib"],
)

py_test(
name = "test_py_obj_scanner",
size = "small",
srcs = dag_tests_srcs,
tags = ["exclusive", "team:core", "ray_dag_tests"],
tags = [
"exclusive",
"ray_dag_tests",
"team:core",
],
deps = [":dag_lib"],
)

py_test_module_list(
files = [
"tests/experimental/test_torch_tensor_dag.py",
"tests/experimental/test_detect_deadlock_dag.py",
],
size = "medium",
tags = ["exclusive", "accelerated_dag", "no_windows", "team:core"],
deps = ["//:ray_lib"],
size = "medium",
files = [
"tests/experimental/test_detect_deadlock_dag.py",
"tests/experimental/test_torch_tensor_dag.py",
],
tags = [
"accelerated_dag",
"exclusive",
"no_windows",
"team:core",
],
deps = ["//:ray_lib"],
)

py_test_module_list(
files = [
"tests/experimental/test_accelerated_dag.py",
],
size = "medium",
tags = ["exclusive", "accelerated_dag", "no_windows", "team:core", "manual"],
deps = ["//:ray_lib"],
size = "medium",
files = [
"tests/experimental/test_accelerated_dag.py",
],
tags = [
"accelerated_dag",
"exclusive",
"manual",
"no_windows",
"team:core",
],
deps = ["//:ray_lib"],
)

py_test(
name = "test_torch_tensor_dag_gpu",
srcs = [
"tests/experimental/test_torch_tensor_dag.py",
],
main = "tests/experimental/test_torch_tensor_dag.py",
size = "medium",
tags = ["exclusive", "accelerated_dag", "no_windows", "team:core", "multi_gpu"],
env = {"RAY_PYTEST_USE_GPU": "1"},
deps = ["//:ray_lib"],
name = "test_torch_tensor_dag_gpu",
size = "medium",
srcs = [
"tests/experimental/test_torch_tensor_dag.py",
],
env = {"RAY_PYTEST_USE_GPU": "1"},
main = "tests/experimental/test_torch_tensor_dag.py",
tags = [
"accelerated_dag",
"exclusive",
"multi_gpu",
"no_windows",
"team:core",
],
deps = ["//:ray_lib"],
)

0 comments on commit 560d968

Please sign in to comment.