WIP: Add explicit Python package requirements

larq · Jun 14, 2024 · daef10c · daef10c
1 parent c7c8c62
commit daef10c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/larq_compute_engine/tflite/python/BUILD b/larq_compute_engine/tflite/python/BUILD
@@ -1,5 +1,6 @@
 load("@org_tensorflow//tensorflow:tensorflow.bzl", "pybind_extension")
 load("@org_tensorflow//tensorflow/lite:build_def.bzl", "tflite_linkopts")
+load("@pypi//:requirements.bzl", "requirement")
 
 package(
     default_visibility = ["//visibility:public"],
@@ -35,6 +36,9 @@ py_library(
         "__init__.py",
         "interpreter_base.py",
     ],
+    deps = [
+        requirement("numpy"),
+    ],
 )
 
 py_library(

diff --git a/larq_compute_engine/tflite/tests/BUILD b/larq_compute_engine/tflite/tests/BUILD
@@ -1,3 +1,5 @@
+load("@pypi//:requirements.bzl", "requirement")
+
 package(
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],  # Apache 2.0
@@ -80,6 +82,7 @@ py_test(
     srcs = ["interpreter_test.py"],
     deps = [
         "//larq_compute_engine/tflite/python:interpreter",
+        requirement("numpy"),
     ],
 )