From c68f4bf21ca372eaf0048ccc98bf99af27667a14 Mon Sep 17 00:00:00 2001
From: Dan Moldovan <mdan@google.com>
Date: Tue, 28 Jan 2020 08:59:36 -0800
Subject: [PATCH] Add a blank py_strict_library that expands to the normal
py_library. Use it in autograph/BUILD.
PiperOrigin-RevId: 291943552
Change-Id: I782be6e676f5d0833b46df11b6263530873b60d0
---
tensorflow/python/autograph/BUILD | 7 +++++--
tensorflow/tensorflow.bzl | 4 ++++
2 files changed, 9 insertions(+), 2 deletions(-)
diff --git a/tensorflow/python/autograph/BUILD b/tensorflow/python/autograph/BUILD
index bdb16ce35f8..874b99464d8 100644
--- a/tensorflow/python/autograph/BUILD
+++ b/tensorflow/python/autograph/BUILD
@@ -1,3 +1,5 @@
+load("//tensorflow:tensorflow.bzl", "py_strict_library")
+
package(
licenses = ["notice"], # Apache 2.0
)
@@ -14,7 +16,7 @@ filegroup(
visibility = ["//tensorflow:__subpackages__"],
)
-py_library(
+py_strict_library(
name = "autograph",
srcs = [
"__init__.py",
@@ -23,9 +25,10 @@ py_library(
visibility = ["//visibility:public"],
deps = [
"//tensorflow/python:util",
+ "//tensorflow/python/autograph/core",
"//tensorflow/python/autograph/impl",
"//tensorflow/python/autograph/lang",
- "//tensorflow/python/autograph/pyct",
+ "//tensorflow/python/autograph/operators",
"//tensorflow/python/autograph/utils",
],
)
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl
index a69309bb281..90ed0957902 100644
--- a/tensorflow/tensorflow.bzl
+++ b/tensorflow/tensorflow.bzl
@@ -1888,6 +1888,10 @@ register_extension_info(
label_regex_for_dep = "{extension_name}",
)
+# Placeholder to use until bazel supports py_strict_library.
+def py_strict_library(name, **kwargs):
+ native.py_library(name = name, **kwargs)
+
def tf_custom_op_py_library(
name,
srcs = [],