Merge pull request #2 from vivarium-collective/update-schema

Update schema
vivarium-collective · Feb 8, 2024 · 0f1c3df · 0f1c3df
2 parents c4f8cfd + 278c22c
commit 0f1c3df
Show file tree

Hide file tree

Showing 4 changed files with 439 additions and 2 deletions.
diff --git a/bigraph_viz/__init__.py b/bigraph_viz/__init__.py
@@ -1,3 +1,17 @@
+import pprint
 from bigraph_viz.plot import plot_bigraph, plot_flow, plot_multitimestep
 from bigraph_viz.dict_utils import pp, pf, schema_state_to_dict
 from bigraph_viz.convert import convert_vivarium_composite
+
+
+pretty = pprint.PrettyPrinter(indent=2)
+
+
+def pp(x):
+    """Print ``x`` in a pretty format."""
+    pretty.pprint(x)
+
+
+def pf(x):
+    """Format ``x`` for display."""
+    return pretty.pformat(x)