Experiment using Bazel's pip_install rule to install keras_applications.
PiperOrigin-RevId: 214076591
diff --git a/WORKSPACE b/WORKSPACE
index 1796182..1160587 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -9,11 +9,27 @@
"https://github.com/bazelbuild/rules_closure/archive/dbb96841cc0a5fb2664c37822803b06dab20c7d1.tar.gz", # 2018-04-13
],
)
-
load("@io_bazel_rules_closure//closure:defs.bzl", "closure_repositories")
-
closure_repositories()
+http_archive(
+ name = "io_bazel_rules_python",
+ strip_prefix = "rules_python-8b5d0683a7d878b28fffe464779c8a53659fc645",
+ urls = [
+ "https://github.com/bazelbuild/rules_python/archive/8b5d0683a7d878b28fffe464779c8a53659fc645.tar.gz",
+ ],
+)
+load("@io_bazel_rules_python//python:pip.bzl", "pip_repositories")
+pip_repositories()
+
+load("@io_bazel_rules_python//python:pip.bzl", "pip_import")
+pip_import(
+ name = "pip_deps",
+ requirements = "//tensorflow:requirements.txt",
+)
+load("@pip_deps//:requirements.bzl", "pip_install")
+pip_install()
+
# We must check the bazel version before trying to parse any other BUILD
# files, in case the parsing of those build files depends on the bazel
# version we require here.
diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD
index 4a72c4b..ac011a2 100755
--- a/tensorflow/python/keras/BUILD
+++ b/tensorflow/python/keras/BUILD
@@ -7,6 +7,7 @@
package(default_visibility = ["//visibility:public"])
+load("@pip_deps//:requirements.bzl", "requirement")
load("//tensorflow:tensorflow.bzl", "py_test")
load("//tensorflow:tensorflow.bzl", "cuda_py_test")
@@ -62,6 +63,7 @@
":backend",
":engine",
":layers",
+ requirement("keras_applications"),
"//tensorflow/python/saved_model",
"//tensorflow/python:training",
],
diff --git a/tensorflow/requirements.txt b/tensorflow/requirements.txt
new file mode 100644
index 0000000..6e111ed
--- /dev/null
+++ b/tensorflow/requirements.txt
@@ -0,0 +1,2 @@
+keras_applications >= 1.0.5
+keras_preprocessing >= 1.0.3