Update tensorboard magic

Author: lukasmasuch

README.md

@@ -692,7 +692,7 @@ If you have opened a Tensorboard instance in a valid log directory, you will see
 If you prefer to see the tensorboard directly within your notebook, you can make use of following **Jupyter magic**:
 
 ```
-%load_ext tensorboard.notebook
+%load_ext tensorboard
 %tensorboard --logdir /workspace/path/to/logs
 ```
 

resources/jupyter/tensorboard_notebook_patch.py

@@ -1,26 +1,28 @@
-"""%tensorboard line magic that patches TensorBoard's implementation to make use of Jupyter 
+"""%tensorboard line magic that patches TensorBoard's implementation to make use of Jupyter
 TensorBoard server extension providing built-in proxying.
 Use:
-    %load_ext tensorboard.notebook
+    %load_ext tensorboard
     %tensorboard --logdir /logs
 """
 
 import argparse
 import uuid
 
-from IPython.display import display, HTML, Javascript
+from IPython.display import HTML, display
+
 
 def _tensorboard_magic(line):
     """Line magic function.
+
     Makes an AJAX call to the Jupyter TensorBoard server extension and outputs
     an IFrame displaying the TensorBoard instance.
     """
     parser = argparse.ArgumentParser()
-    parser.add_argument('--logdir', default='/workspace/environment/')
+    parser.add_argument("--logdir", default="/workspace/")
     args = parser.parse_args(line.split())
-    
-    iframe_id = 'tensorboard-' + str(uuid.uuid4())
-        
+
+    iframe_id = "tensorboard-" + str(uuid.uuid4())
+
     html = """
 <!-- JUPYTER_TENSORBOARD_TEST_MARKER -->
 <script>
@@ -42,23 +44,37 @@ def _tensorboard_magic(line):
     style="width: 100%%; height: 620px; display: none;"
     frameBorder="0">
 </iframe>
-""" % (args.logdir, iframe_id, iframe_id)
-    
+""" % (
+        args.logdir,
+        iframe_id,
+        iframe_id,
+    )
+
     display(HTML(html))
-    
+
+
 def load_ipython_extension(ipython):
-    """IPython extension entry point."""
-    ipython.register_magic_function(
-        _tensorboard_magic,
-        magic_kind='line',
-        magic_name='tensorboard',
+    """Deprecated: use `%load_ext tensorboard` instead.
+
+    Raises:
+      RuntimeError: Always.
+    """
+    raise RuntimeError(
+        "Use '%load_ext tensorboard' instead of '%load_ext tensorboard.notebook'."
     )
 
+
 def _load_ipython_extension(ipython):
     """Load the TensorBoard notebook extension.
+
     Intended to be called from `%load_ext tensorboard`. Do not invoke this
     directly.
+
     Args:
       ipython: An `IPython.InteractiveShell` instance.
     """
-    load_ipython_extension(ipython)
+    ipython.register_magic_function(
+        _tensorboard_magic,
+        magic_kind="line",
+        magic_name="tensorboard",
+    )