From a06a84787ca0fee88441736cae106c7a05885b9a Mon Sep 17 00:00:00 2001
From: Alexander Viand <alexander.viand@intel.com>
Date: Wed, 29 Jan 2025 11:57:10 -0800
Subject: [PATCH] bazel py_test fixes

---
 frontend/BUILD       | 1 +
 frontend/testing.bzl | 1 +
 2 files changed, 2 insertions(+)

diff --git a/frontend/BUILD b/frontend/BUILD
index 190aee59b..342bf2486 100644
--- 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
index 3d24a318c..902f7dd60 100644
--- 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 = {