updated model graphs and distributed model tutorial

AthenaEPI · Aug 21, 2018 · f635d70 · f635d70
1 parent 1d49c89
commit f635d70
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 11 deletions.
diff --git a/dmipy/core/modeling_framework.py b/dmipy/core/modeling_framework.py
@@ -703,7 +703,8 @@ def _add_optimization_parameter(
             {(None, 'fraction'): parameter_name})
 
     def visualize_model_setup(
-            self, view=True, cleanup=True, with_parameters=False):
+            self, view=True, cleanup=True, with_parameters=False,
+            im_format='png'):
         """
         Visualizes MultiCompartmentModel setup using graphviz module. It uses
         the uuid module to create a unique identifier for each model in the
@@ -727,7 +728,7 @@ def visualize_model_setup(
         with_parameters: boolean,
             Whether or not to also visualize the parameters of each model.
         """
-        dot = Digraph('Model Setup')
+        dot = Digraph('Model Setup', format=im_format)
         base_model = self.__class__.__name__
         base_uuid = str(uuid4())
         dot.node(base_uuid, base_model)

diff --git a/examples/tutorial_distributed_model_representations.ipynb b/examples/tutorial_distributed_model_representations.ipynb