refactor: copy is notebook function from hubble sdk

Signed-off-by: anna-charlotte <[email protected]>
docarray · anna-charlotte · Feb 16, 2023 · Feb 9, 2023 · Feb 9, 2023 · Feb 9, 2023
commit 05d8461b94ad1074392a0d7fa19a975ddf783045
diff --git a/docarray/typing/url/url_3d/point_cloud_url.py b/docarray/typing/url/url_3d/point_cloud_url.py
@@ -7,6 +7,7 @@
 from docarray.typing.tensor.abstract_tensor import AbstractTensor
 from docarray.typing.tensor.ndarray import NdArray
 from docarray.typing.url.url_3d.url_3d import Url3D
+from docarray.utils.misc import is_notebook
 
 T = TypeVar('T', bound='PointCloud3DUrl')
 
@@ -82,7 +83,6 @@ def _display_point_cloud(
         shape (n_points, 3) or (n_points, 4).
     """
     import trimesh
-    from hubble.utils.notebook import is_notebook
     from IPython.display import display
 
     if colors is None:

diff --git a/docarray/utils/misc.py b/docarray/utils/misc.py
@@ -20,3 +20,29 @@ def is_torch_available():
 
 def is_tf_available():
     return tf_imported
+
+
+def is_notebook() -> bool:
+    """
+    Check if we're running in a Jupyter notebook, using magic command
+    `get_ipython` that only available in Jupyter.
+
+    :return: True if run in a Jupyter notebook else False.
+    """
+
+    try:
+        shell = get_ipython().__class__.__name__  # noqa: F821
+    except NameError:
+        return False
+
+    if shell == 'ZMQInteractiveShell':
+        return True
+
+    elif shell == 'Shell':
+        return True
+
+    elif shell == 'TerminalInteractiveShell':
+        return False
+
+    else:
+        return False