Limit visibility of Reverb's dataset targets.

PiperOrigin-RevId: 358164326 Change-Id: Ife381d922d517a44dc2411f90ddae1f3714c3282
google-deepmind · Feb 18, 2021 · 909c27f · 909c27f
1 parent 834d4ce
commit 909c27f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/reverb/BUILD b/reverb/BUILD
@@ -134,6 +134,7 @@ reverb_pytype_library(
     srcs = ["dataset.py"],
     srcs_version = "PY3",
     strict_deps = True,
+    visibility = [":__subpackages__"],
     deps = [
         ":client",
         ":replay_sample",
@@ -146,6 +147,7 @@ reverb_pytype_library(
     srcs = ["trajectory_dataset.py"],
     srcs_version = "PY3",
     strict_deps = True,
+    visibility = [":__subpackages__"],
     deps = [
         ":client",
         ":replay_sample",
@@ -158,6 +160,7 @@ reverb_pytype_library(
     srcs = ["timestep_dataset.py"],
     srcs_version = "PY3",
     strict_deps = True,
+    visibility = [":__subpackages__"],
     deps = [
         ":client",
         ":replay_sample",