Add py load statement for codegen (#2166)

The new codegen python build rules were added without properly loading the rules. BUG=b/295216390
tensorflow · Aug 14, 2023 · e3d1a5a · e3d1a5a
1 parent feb2669
commit e3d1a5a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/codegen/BUILD b/codegen/BUILD
@@ -1,3 +1,4 @@
+load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 load("@tflm_pip_deps//:requirements.bzl", "requirement")
 
 package(