Restore event_processing APIs (#378)

Many teams depend on those APIs internally.
tensorflow · Aug 18, 2017 · c1be95b · c1be95b
1 parent 7fe6f60
commit c1be95b
Show file tree

Hide file tree

Showing 26 changed files with 2,904 additions and 278 deletions.
diff --git a/tensorboard/backend/application.py b/tensorboard/backend/application.py
@@ -37,8 +37,8 @@
 
 from tensorboard import db
 from tensorboard.backend import http_util
-from tensorboard.backend.event_processing import event_accumulator
-from tensorboard.backend.event_processing import event_multiplexer
+from tensorboard.backend.event_processing import plugin_event_accumulator as event_accumulator  # pylint: disable=line-too-long
+from tensorboard.backend.event_processing import plugin_event_multiplexer as event_multiplexer  # pylint: disable=line-too-long
 from tensorboard.plugins import base_plugin
 from tensorboard.plugins.audio import metadata as audio_metadata
 from tensorboard.plugins.core import core_plugin

diff --git a/tensorboard/backend/application_test.py b/tensorboard/backend/application_test.py
@@ -33,7 +33,7 @@
 
 from tensorboard import main as tensorboard
 from tensorboard.backend import application
-from tensorboard.backend.event_processing import event_multiplexer
+from tensorboard.backend.event_processing import plugin_event_multiplexer as event_multiplexer  # pylint: disable=line-too-long
 from tensorboard.plugins import base_plugin
 
 

diff --git a/tensorboard/backend/event_processing/BUILD b/tensorboard/backend/event_processing/BUILD
@@ -74,7 +74,10 @@ py_test(
 
 py_library(
     name = "event_accumulator",
-    srcs = ["event_accumulator.py"],
+    srcs = [
+        "event_accumulator.py",
+        "plugin_event_accumulator.py",
+    ],
     srcs_version = "PY2AND3",
     visibility = ["//visibility:public"],
     deps = [
@@ -103,9 +106,27 @@ py_test(
     ],
 )
 
+py_test(
+    name = "plugin_event_accumulator_test",
+    size = "small",
+    srcs = ["plugin_event_accumulator_test.py"],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":event_accumulator",
+        "//tensorboard:expect_tensorflow_installed",
+        "//tensorboard/plugins/audio:summary",
+        "//tensorboard/plugins/distribution:compressor",
+        "//tensorboard/plugins/image:summary",
+        "//tensorboard/plugins/scalar:summary",
+    ],
+)
+
 py_library(
     name = "event_multiplexer",
-    srcs = ["event_multiplexer.py"],
+    srcs = [
+        "event_multiplexer.py",
+        "plugin_event_multiplexer.py",
+    ],
     srcs_version = "PY2AND3",
     visibility = ["//visibility:public"],
     deps = [
@@ -129,6 +150,18 @@ py_test(
     ],
 )
 
+py_test(
+    name = "plugin_event_multiplexer_test",
+    size = "small",
+    srcs = ["plugin_event_multiplexer_test.py"],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":event_accumulator",
+        ":event_multiplexer",
+        "//tensorboard:expect_tensorflow_installed",
+    ],
+)
+
 py_library(
     name = "plugin_asset_util",
     srcs = ["plugin_asset_util.py"],