Explain how to make the graph from to_graphviz

Author: Armavica

pymc/model/core.py

@@ -1903,7 +1903,7 @@ def to_graphviz(
         .. code-block:: python
 
             import numpy as np
-            from pymc import HalfCauchy, Model, Normal, model_to_graphviz
+            from pymc import HalfCauchy, Model, Normal
 
             J = 8
             y = np.array([28, 8, -3, 7, -1, 1, 18, 12])
@@ -1920,6 +1920,15 @@ def to_graphviz(
                 obs = Normal("obs", theta, sigma=sigma, observed=y)
 
             schools.to_graphviz()
+
+        Note that this code automatically plots the graph if executed in a Jupyter notebook.
+        If executed non-interactively, such as in a script or python console, the graph
+        needs to be rendered explicitly:
+
+        .. code-block:: python
+
+            # creates the file `schools.pdf`
+            schools.to_graphviz().render("schools")
         """
         return model_to_graphviz(
             model=self,

pymc/model_graph.py

@@ -474,6 +474,15 @@ def model_to_graphviz(
             obs = Normal("obs", theta, sigma=sigma, observed=y)
 
         model_to_graphviz(schools)
+
+    Note that this code automatically plots the graph if executed in a Jupyter notebook.
+    If executed non-interactively, such as in a script or python console, the graph
+    needs to be rendered explicitly:
+
+    .. code-block:: python
+
+        # creates the file `schools.pdf`
+        model_to_graphviz(schools).render("schools")
     """
     if "plain" not in formatting:
         raise ValueError(f"Unsupported formatting for graph nodes: '{formatting}'. See docstring.")