Integrate with bigraph-schema

bigraph_viz/__init__.py

@@ -1,3 +1,11 @@
-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
+import pprint
+from bigraph_viz.diagram import plot_bigraph
+from bigraph_viz.dict_utils import replace_regex_recursive
+from bigraph_schema import TypeSystem
+
+
+pretty = pprint.PrettyPrinter(indent=2)
+
+
+def pf(x):
+    return pretty.pformat(x)

bigraph_viz/convert.py

@@ -6,7 +6,7 @@
 
 import copy
 
-from bigraph_viz import plot_bigraph, pp
+from bigraph_viz import plot_bigraph
 from bigraph_viz.dict_utils import deep_merge, absolute_path, nest_path