bazel py_test fixes

google · Jan 29, 2025 · a06a847 · a06a847
1 parent 359a68f
commit a06a847
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/frontend/BUILD b/frontend/BUILD
@@ -34,6 +34,7 @@ py_library(
     ),
     data = DATA_DEPS,
     deps = [
+        "@frontend_pip_deps_colorama//:pkg",
         "@frontend_pip_deps_numba//:pkg",
         "@frontend_pip_deps_pybind11//:pkg",
         "@frontend_pip_deps_pybind11_global//:pkg",

diff --git a/frontend/testing.bzl b/frontend/testing.bzl
@@ -29,6 +29,7 @@ def frontend_test(name, srcs, deps = [], data = [], tags = []):
             ":frontend",
             "@com_google_absl_py//absl/testing:absltest",
         ],
+        imports = ["."],
         data = data,
         tags = tags,
         env = {