Jimmy-INL
diff --git a/‎.circleci/config.yml
Lines changed: 2 additions & 0 deletions b/‎.circleci/config.yml
Lines changed: 2 additions & 0 deletions
diff --git a/‎README.md
Lines changed: 1 addition & 8 deletions b/‎README.md
Lines changed: 1 addition & 8 deletions
diff --git a/‎RELEASE.md
Lines changed: 4 additions & 0 deletions b/‎RELEASE.md
Lines changed: 4 additions & 0 deletions
diff --git a/‎causalnex/__init__.py
Lines changed: 1 addition & 1 deletion b/‎causalnex/__init__.py
Lines changed: 1 addition & 1 deletion
diff --git a/‎causalnex/plots/__init__.py
Lines changed: 2 additions & 11 deletions b/‎causalnex/plots/__init__.py
Lines changed: 2 additions & 11 deletions
diff --git a/‎causalnex/plots/display.py
Lines changed: 39 additions & 82 deletions b/‎causalnex/plots/display.py
Lines changed: 39 additions & 82 deletions
@@ -24,6 +24,7 @@ utils:
       conda create --name=causalnex_env python=${PYTHON_VERSION} -y
       conda activate causalnex_env
       conda install -y virtualenv
+      pip install cython
       pip install -U pip setuptools wheel
       sudo apt-get install graphviz
 
@@ -182,6 +183,7 @@ jobs:
           name: Success!
           command: echo "All checks passed"
 
+
 ###############################################################################
 workflows:
   version: 2
 
@@ -52,14 +52,7 @@ CausalNex is a Python package. To install it, simply run:
 pip install causalnex
 ```
 
-Since pygraphviz can be difficult to install, esp. on Windows machines, the requirement is optional.
-If you want to use the causalnex native plotting tools, you can use
-```bash
-pip install "causalnex[plot]"
-```
-Alternatively, you can use the `networkx` drawing functionality for visualisations with fewer dependencies.
-
-Use `all` for a full installation of dependencies (only the plotting right now):
+Use `all` for a full installation of dependencies:
 ```bash
 pip install "causalnex[all]"
 ```
 
@@ -1,4 +1,8 @@
 # Upcoming release
+# Release 0.12.0
+* Switch to Pyvis for graph visualisation and remove dependency on Graphviz
+
+# Release 0.11.2
 * Support newer version of Numpy
 * Support newer version of Scikit-learn
 * Remove python 3.6, 3.7 support
 
@@ -30,6 +30,6 @@
 causalnex toolkit for causal reasoning (Bayesian Networks / Inference)
 """
 
-__version__ = "0.11.2"
+__version__ = "0.12.0"
 
 __all__ = ["structure", "discretiser", "evaluation", "inference", "network", "plots"]
@@ -29,22 +29,13 @@
 """
 ``causalnex.plots`` provides functionality to visualise structure models.
 """
-
 __all__ = [
     "plot_structure",
     "NODE_STYLE",
     "EDGE_STYLE",
     "GRAPH_STYLE",
-    "color_gradient_string",
     "display_plot_ipython",
-    "display_plot_mpl",
 ]
 
-from .display import display_plot_ipython, display_plot_mpl
-from .plots import (
-    EDGE_STYLE,
-    GRAPH_STYLE,
-    NODE_STYLE,
-    color_gradient_string,
-    plot_structure,
-)
+from .display import display_plot_ipython
+from .plots import EDGE_STYLE, GRAPH_STYLE, NODE_STYLE, plot_structure
@@ -25,97 +25,54 @@
 #
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""Methods to display styled pygraphgiz plots."""
+"""Methods to display styled pyvis plots."""
+import json
+from html import escape
+from typing import Dict
 
-import io
-from typing import Any, Tuple
+from IPython.lib.display import IFrame
+from pyvis.network import Network
 
-try:
-    from pygraphviz.agraph import AGraph
-except ImportError:
-    AGraph = Any
 
-try:
-    from IPython.display import Image
-except ImportError:
-    Image = Any
-
-try:
-    import matplotlib.pyplot as plt
-    from matplotlib.axes import Axes
-    from matplotlib.figure import Figure
-except ImportError:
-    Axes = Any
-    Figure = Any
-
-
-def display_plot_ipython(viz: AGraph, prog: str = "neato") -> Image:
+def display_plot_ipython(
+    viz: Network,
+    output_filename: str,
+    layout_kwargs: Dict[str, Dict] = None,
+) -> IFrame:
     """
-    Displays a pygraphviz object using ipython.
-
+    Displays a PyVis object using ipython.
     Args:
-        viz: pygraphviz object to render.
-
-        prog: The graph layout. Avaliable are:
-        dot, neato, fdp, sfdp, twopi and circo
-
+        viz: pyvis object to render.
+        output_filename: write html to a given path, e.g. "./plot.html". File as to end in ".html"
+        layout_kwargs: Dictionary to set the `layout` and `physics` of the graph.
+            Example:
+            ::
+                >>> layout_kwargs = {
+                        "physics": {
+                            "solver": "repulsion"
+                            },
+                        "layout": {
+                            "hierarchical": {
+                                "enabled": True
+                                }
+                            }
+                        }
     Returns:
-        IPython Image object. Renders in a notebook.
-
+        IPython IFrame object. Renders in a notebook.
     Raises:
         ImportError: if IPython is not installed (optional dependency).
     """
+    layout_kwargs = layout_kwargs or {}
+    viz.set_options(options=json.dumps(layout_kwargs))
 
-    if Image is Any:
-        raise ImportError("display_plot_ipython method requires IPython installed.")
-
-    return Image(viz.draw(format="png", prog=prog))
-
-
-def display_plot_mpl(
-    viz: AGraph,
-    prog: str = "neato",
-    ax: Axes = None,
-    pixel_size_in: float = 0.01,
-) -> Tuple[Figure, Axes]:
-    """
-    Displays a pygraphviz object using matplotlib.
-
-    Args:
-        viz: pygraphviz object to render.
-
-        prog: The graph layout. Avaliable are:
-        dot, neato, fdp, sfdp, twopi and circo
-
-        ax: Optional matplotlib axes to plot on.
-
-        pixel_size_in: Scaling multiple for the plot.
-
-    Returns:
-        IPython Image object. Renders in a notebook.
-
-    Raises:
-        ImportError: if matplotlib is not installed (optional dependency).
-    """
-
-    if Figure is Any:
-        raise ImportError("display_plot_mpl method requires matplotlib installed.")
-
-    # bytes:
-    s = viz.draw(format="png", prog=prog)
-    # convert to numpy array
-    array = plt.imread(io.BytesIO(s))
-    x_dim, y_dim, _ = array.shape
+    html = viz.generate_html()
 
-    # handle passed axis
-    if ax is not None:
-        ax.imshow(array)
-        ax.axis("off")
-        return None, ax
+    with open(output_filename, mode="w", encoding="UTF-8") as f:
+        f.write(html)
 
-    # handle new axis
-    f, ax = plt.subplots(1, 1, figsize=(y_dim * pixel_size_in, x_dim * pixel_size_in))
-    ax.imshow(array)
-    ax.axis("off")
-    f.tight_layout(pad=0.0)
-    return f, ax
+    return IFrame(
+        src=output_filename,
+        width=viz.width,
+        height=viz.height,
+        extras=[f"srcdoc='{escape(viz.generate_html(notebook=False))}'"],
+    )