diff options
author | Michael Case <mikecase@google.com> | 2018-09-21 18:56:21 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 18:59:56 -0700 |
commit | 7229d08f0b25e24e6dd4833a94a27f404b27a350 (patch) | |
tree | a42dcae0372769059c7e3e88ef6b50944c46c5a9 | |
parent | a2fd40adcc714f18167acd9650e5442d4afd6a01 (diff) |
Experiment using Bazel's pip_install rule to install keras_applications.
PiperOrigin-RevId: 214076591
-rw-r--r-- | WORKSPACE | 20 | ||||
-rwxr-xr-x | tensorflow/python/keras/BUILD | 2 | ||||
-rw-r--r-- | tensorflow/requirements.txt | 2 |
3 files changed, 22 insertions, 2 deletions
@@ -9,11 +9,27 @@ http_archive( "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 4a72c4b3f3..ac011a2940 100755 --- a/tensorflow/python/keras/BUILD +++ b/tensorflow/python/keras/BUILD @@ -7,6 +7,7 @@ exports_files(["LICENSE"]) 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 @@ py_library( ":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 0000000000..6e111edefc --- /dev/null +++ b/tensorflow/requirements.txt @@ -0,0 +1,2 @@ +keras_applications >= 1.0.5 +keras_preprocessing >= 1.0.3 |