Minor updates to build system.

PiperOrigin-RevId: 437864517
Change-Id: I941bf37463410af6965b29e98dcb7858adcb83dc

Author: Johannes Ballé

BUILD

@@ -1,5 +1,5 @@
 package(
-    default_visibility = ["//visibility:private"],
+    default_visibility = ["//visibility:public"],
 )
 
 licenses(["notice"])
@@ -9,7 +9,6 @@ exports_files(["LICENSE"])
 py_library(
     name = "tensorflow_compression",
     srcs = ["tensorflow_compression/__init__.py"],
-    srcs_version = "PY3",
     visibility = ["//visibility:public"],
     deps = [
         "//tensorflow_compression/python/datasets:y4m_dataset",
@@ -33,6 +32,11 @@ py_library(
     ],
 )
 
+py_binary(
+    name = "build_docs",
+    srcs = ["build_docs.py"],
+)
+
 py_binary(
     name = "build_pip_pkg",
     srcs = ["build_pip_pkg.py"],
@@ -51,5 +55,4 @@ py_binary(
         "//tensorflow_compression/python/ops:py_src",
         "//tensorflow_compression/python/util:py_src",
     ],
-    python_version = "PY3",
 )

tensorflow_compression/cc/BUILD

@@ -1,3 +1,5 @@
+package(default_visibility = ["//visibility:public"])
+
 licenses(["notice"])  # Apache 2.0
 
 cc_binary(

tensorflow_compression/python/datasets/BUILD

@@ -7,14 +7,12 @@ licenses(["notice"])
 py_library(
     name = "y4m_dataset",
     srcs = ["y4m_dataset.py"],
-    srcs_version = "PY3",
     deps = ["//tensorflow_compression/python/ops:gen_ops"],
 )
 
 py_test(
     name = "y4m_dataset_test",
     srcs = ["y4m_dataset_test.py"],
-    python_version = "PY3",
     deps = [":y4m_dataset"],
 )
 

tensorflow_compression/python/distributions/BUILD

@@ -7,7 +7,6 @@ licenses(["notice"])
 py_library(
     name = "deep_factorized",
     srcs = ["deep_factorized.py"],
-    srcs_version = "PY3",
     deps = [
         ":helpers",
         ":uniform_noise",
@@ -17,7 +16,6 @@ py_library(
 py_test(
     name = "deep_factorized_test",
     srcs = ["deep_factorized_test.py"],
-    python_version = "PY3",
     deps = [
         ":deep_factorized",
         ":helpers",
@@ -27,13 +25,11 @@ py_test(
 py_library(
     name = "helpers",
     srcs = ["helpers.py"],
-    srcs_version = "PY3",
 )
 
 py_test(
     name = "helpers_test",
     srcs = ["helpers_test.py"],
-    python_version = "PY3",
     deps = [
         ":deep_factorized",
         ":helpers",
@@ -43,14 +39,12 @@ py_test(
 py_library(
     name = "uniform_noise",
     srcs = ["uniform_noise.py"],
-    srcs_version = "PY3",
     deps = [":helpers"],
 )
 
 py_test(
     name = "uniform_noise_test",
     srcs = ["uniform_noise_test.py"],
-    python_version = "PY3",
     deps = [
         ":helpers",
         ":uniform_noise",
@@ -60,7 +54,6 @@ py_test(
 py_library(
     name = "round_adapters",
     srcs = ["round_adapters.py"],
-    srcs_version = "PY3",
     deps = [
         ":deep_factorized",
         ":helpers",
@@ -72,7 +65,6 @@ py_library(
 py_test(
     name = "round_adapters_test",
     srcs = ["round_adapters_test.py"],
-    python_version = "PY3",
     deps = [
         ":deep_factorized",
         ":round_adapters",

tensorflow_compression/python/entropy_models/BUILD

@@ -7,7 +7,6 @@ licenses(["notice"])
 py_library(
     name = "continuous_base",
     srcs = ["continuous_base.py"],
-    srcs_version = "PY3",
     deps = [
         "//tensorflow_compression/python/distributions:helpers",
         "//tensorflow_compression/python/ops:gen_ops",
@@ -17,7 +16,6 @@ py_library(
 py_library(
     name = "continuous_batched",
     srcs = ["continuous_batched.py"],
-    srcs_version = "PY3",
     deps = [
         ":continuous_base",
         "//tensorflow_compression/python/distributions:helpers",
@@ -30,7 +28,6 @@ py_library(
 py_test(
     name = "continuous_batched_test",
     srcs = ["continuous_batched_test.py"],
-    python_version = "PY3",
     deps = [
         ":continuous_batched",
         "//tensorflow_compression/python/distributions:uniform_noise",
@@ -40,7 +37,6 @@ py_test(
 py_library(
     name = "continuous_indexed",
     srcs = ["continuous_indexed.py"],
-    srcs_version = "PY3",
     deps = [
         ":continuous_base",
         "//tensorflow_compression/python/ops:gen_ops",
@@ -53,7 +49,6 @@ py_test(
     name = "continuous_indexed_test",
     timeout = "long",
     srcs = ["continuous_indexed_test.py"],
-    python_version = "PY3",
     shard_count = 6,
     deps = [
         ":continuous_indexed",
@@ -64,7 +59,6 @@ py_test(
 py_library(
     name = "universal",
     srcs = ["universal.py"],
-    srcs_version = "PY3",
     deps = [
         ":continuous_base",
         "//tensorflow_compression/python/ops:gen_ops",
@@ -75,7 +69,6 @@ py_library(
 py_test(
     name = "universal_test",
     srcs = ["universal_test.py"],
-    python_version = "PY3",
     shard_count = 3,
     deps = [
         ":universal",

tensorflow_compression/python/layers/BUILD

@@ -7,14 +7,12 @@ licenses(["notice"])
 py_library(
     name = "gdn",
     srcs = ["gdn.py"],
-    srcs_version = "PY3",
     deps = [":parameters"],
 )
 
 py_test(
     name = "gdn_test",
     srcs = ["gdn_test.py"],
-    python_version = "PY3",
     deps = [
         ":gdn",
         ":parameters",
@@ -24,34 +22,29 @@ py_test(
 py_library(
     name = "initializers",
     srcs = ["initializers.py"],
-    srcs_version = "PY3",
 )
 
 py_test(
     name = "initializers_test",
     srcs = ["initializers_test.py"],
-    python_version = "PY3",
     deps = [":initializers"],
 )
 
 py_library(
     name = "parameters",
     srcs = ["parameters.py"],
-    srcs_version = "PY3",
     deps = ["//tensorflow_compression/python/ops:math_ops"],
 )
 
 py_test(
     name = "parameters_test",
     srcs = ["parameters_test.py"],
-    python_version = "PY3",
     deps = [":parameters"],
 )
 
 py_library(
     name = "signal_conv",
     srcs = ["signal_conv.py"],
-    srcs_version = "PY3",
     deps = [
         ":parameters",
         "//tensorflow_compression/python/ops:padding_ops",
@@ -62,7 +55,6 @@ py_test(
     name = "signal_conv_test",
     timeout = "long",
     srcs = ["signal_conv_test.py"],
-    python_version = "PY3",
     shard_count = 3,
     deps = [
         ":initializers",
@@ -74,14 +66,12 @@ py_test(
 py_library(
     name = "soft_round",
     srcs = ["soft_round.py"],
-    srcs_version = "PY3",
     deps = ["//tensorflow_compression/python/ops:round_ops"],
 )
 
 py_test(
     name = "soft_round_test",
     srcs = ["soft_round_test.py"],
-    python_version = "PY3",
     deps = [
         ":soft_round",
         "//tensorflow_compression/python/ops:round_ops",

tensorflow_compression/python/ops/BUILD

@@ -8,19 +8,16 @@ py_library(
     name = "gen_ops",
     srcs = ["gen_ops.py"],
     data = ["//tensorflow_compression/cc:libtensorflow_compression.so"],
-    srcs_version = "PY3",
 )
 
 py_library(
     name = "math_ops",
     srcs = ["math_ops.py"],
-    srcs_version = "PY3",
 )
 
 py_test(
     name = "math_ops_test",
     srcs = ["math_ops_test.py"],
-    python_version = "PY3",
     deps = [
         ":math_ops",
         ":round_ops",
@@ -30,33 +27,28 @@ py_test(
 py_library(
     name = "padding_ops",
     srcs = ["padding_ops.py"],
-    srcs_version = "PY3",
 )
 
 py_test(
     name = "padding_ops_test",
     srcs = ["padding_ops_test.py"],
-    python_version = "PY3",
     deps = [":padding_ops"],
 )
 
 py_test(
     name = "range_coding_ops_test",
     srcs = ["range_coding_ops_test.py"],
-    python_version = "PY3",
     deps = [":gen_ops"],
 )
 
 py_library(
     name = "round_ops",
     srcs = ["round_ops.py"],
-    srcs_version = "PY3",
 )
 
 py_test(
     name = "round_ops_test",
     srcs = ["round_ops_test.py"],
-    python_version = "PY3",
     deps = [":round_ops"],
 )
 

tensorflow_compression/python/util/BUILD

@@ -7,13 +7,11 @@ licenses(["notice"])
 py_library(
     name = "packed_tensors",
     srcs = ["packed_tensors.py"],
-    srcs_version = "PY3",
 )
 
 py_test(
     name = "packed_tensors_test",
     srcs = ["packed_tensors_test.py"],
-    python_version = "PY3",
     deps = [":packed_tensors"],
 )