Merge pull request #32 from ferrarodav/patch-1

Patched for tensorflow 1.11
lspvic · Oct 19, 2018 · b6889d3 · b6889d3
2 parents d33eff7 + f6fd86a
commit b6889d3
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/jupyter_tensorboard/tensorboard_manager.py b/jupyter_tensorboard/tensorboard_manager.py
@@ -15,20 +15,19 @@
 try:
     # Tensorboard 0.4.x above series
     from tensorboard import default
-    if hasattr(default, 'PLUGIN_LOADERS'):
-        # Tensorflow 1.10 series
+    if hasattr(default, 'PLUGIN_LOADERS') or hasattr(default, '_PLUGINS'):
+        # Tensorflow 1.10 or above series
         logging.debug("Tensorboard 1.10 or above series detected")
         from tensorboard import program
 
         def create_tb_app(logdir, reload_interval, purge_orphaned_data):
             argv = [
+                        "",
                         "--logdir", logdir,
                         "--reload_interval", str(reload_interval),
                         "--purge_orphaned_data", str(purge_orphaned_data),
                    ]
-            tensorboard = program.TensorBoard(
-                default.PLUGIN_LOADERS,
-                default.get_assets_zip_provider())
+            tensorboard = program.TensorBoard()
             tensorboard.configure(argv)
             return application.standard_tensorboard_wsgi(
                 tensorboard.flags,