From 7229d08f0b25e24e6dd4833a94a27f404b27a350 Mon Sep 17 00:00:00 2001 From: Michael Case Date: Fri, 21 Sep 2018 18:56:21 -0700 Subject: [PATCH] Experiment using Bazel's pip_install rule to install keras_applications. PiperOrigin-RevId: 214076591 --- WORKSPACE | 20 ++++++++++++++++++-- tensorflow/python/keras/BUILD | 2 ++ tensorflow/requirements.txt | 2 ++ 3 files changed, 22 insertions(+), 2 deletions(-) create mode 100644 tensorflow/requirements.txt diff --git a/WORKSPACE b/WORKSPACE index 17961829a60..11605871f34 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -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 4a72c4b3f3e..ac011a2940e 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 00000000000..6e111edefc6 --- /dev/null +++ b/tensorflow/requirements.txt @@ -0,0 +1,2 @@ +keras_applications >= 1.0.5 +keras_preprocessing >= 1.0.3