Setup Sonnet Jenkins tests for py2.

PiperOrigin-RevId: 154536824
google-deepmind · Apr 28, 2017 · 697c20f · 697c20f
1 parent 23972bb
commit 697c20f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/sonnet/python/BUILD b/sonnet/python/BUILD
@@ -175,6 +175,7 @@ module_tests = [
     name = test_name,
     size = test_size,
     srcs = ["modules/%s%s.py" % (test_subdir, test_name)],
+    srcs_version = "PY2AND3",
     deps = [
         # mox dep,
         "//sonnet",

diff --git a/sonnet/tensorflow.bzl b/sonnet/tensorflow.bzl
@@ -81,7 +81,7 @@ def tf_custom_op_library_additional_deps():
   return [
       "@org_tensorflow//third_party/eigen3",
       "@org_tensorflow//tensorflow/core:framework_headers_lib",
-      "@protobuf//:protobuf_headers",
+      "@protobuf//:protobuf",
   ]
 
 def tf_custom_op_library(name, srcs=[], gpu_srcs=[], deps=[]):