howso-visuals 2.1.1__tar.gz → 2.2.0__tar.gz

{howso_visuals-2.1.1 → howso_visuals-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: howso-visuals
-Version: 2.1.1
+Version: 2.2.0
 Summary: Visualization utilities for use with Howso Engine.
 License: GNU AFFERO GENERAL PUBLIC LICENSE
            Version 3, 19 November 2007

{howso_visuals-2.1.1 → howso_visuals-2.2.0}/howso/visuals/graph.py

@@ -103,6 +103,7 @@ def _create_edge_annotations(
 def plot_graph(
     G: nx.Graph,  # noqa: N803
     *,
+    cscale_tuple: tuple[float, float, float] = None,
     edge_attr_sigfigs: SupportsInt | None = 4,
     edge_attr: str | None = None,
     label_edges: bool = True,
@@ -118,6 +119,8 @@ def plot_graph(
     ----------
     G : nx.Graph
         The graph to plot.
+    cscale_tuple : tuple[float, float, float], optional
+        The tuple of values to use for the colorscale. If None, (3, 15, 30) will be used.
     edge_attr : str, optional
         The name of the edge attribute to use when scaling the size of the edges. This should
         be an attribute that is contained within ``G``.
@@ -196,14 +199,28 @@ def plot_graph(
             annotations=annotations,
         )
     )
+    
+    if cscale_tuple is None:
+        cbot = 1
+        cmin = 3
+        cmid = 15
+        cmax = 30
+    else:
+        cbot = 1
+        cmin = cscale_tuple[0]
+        cmid = cscale_tuple[1]
+        cmax = cscale_tuple[2]
+
     fig.update_layout(
         coloraxis=dict(
             colorscale="Bluered_r",
-            cmin=1,
-            cmax=30,
-            cmid=15,
+            cmin=cbot,
+            cmid=cmid,
+            cmax=cmax,
             colorbar=dict(
-                title="Missing Information Ratio", tickvals=[1, 3, 15, 30], ticktext=["1", "3", "15", "≥30"]
+                title="Missing Information Ratio",
+                tickvals=[cbot, cmin, cmid, cmax],
+                ticktext=[f"{cbot}", f"{cmin}", f"{cmid}", f"≥{cmax}"],
             ),
             reversescale=True,
         ),

{howso_visuals-2.1.1 → howso_visuals-2.2.0}/howso_visuals.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: howso-visuals
-Version: 2.1.1
+Version: 2.2.0
 Summary: Visualization utilities for use with Howso Engine.
 License: GNU AFFERO GENERAL PUBLIC LICENSE
            Version 3, 19 November 2007