Experiment using Bazel's pip_install rule to install keras_applications.

PiperOrigin-RevId: 214076591
This commit is contained in:
Michael Case 2018-09-21 18:56:21 -07:00 committed by TensorFlower Gardener
parent a2fd40adcc
commit 7229d08f0b
3 changed files with 22 additions and 2 deletions

View File

@ -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.

View File

@ -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",
],

View File

@ -0,0 +1,2 @@
keras_applications >= 1.0.5
keras_preprocessing >= 1.0.3