phylogenie 2.1.10__tar.gz → 2.1.12__tar.gz

{phylogenie-2.1.10 → phylogenie-2.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: phylogenie
-Version: 2.1.10
+Version: 2.1.12
 Summary: Generate phylogenetic datasets with minimal setup effort
 Author: Gabriele Marino
 Author-email: gabmarino.8601@gmail.com

{phylogenie-2.1.10 → phylogenie-2.1.12}/phylogenie/__init__.py

@@ -12,6 +12,7 @@ from phylogenie.generators import (
 )
 from phylogenie.io import dump_newick, load_fasta, load_newick
 from phylogenie.msa import MSA
+from phylogenie.plot import Coloring, plot_tree
 from phylogenie.skyline import (
     SkylineMatrix,
     SkylineMatrixCoercible,
@@ -90,4 +91,6 @@ __all__ = [
     "load_fasta",
     "load_newick",
     "MSA",
+    "Coloring",
+    "plot_tree",
 ]

{phylogenie-2.1.10 → phylogenie-2.1.12}/phylogenie/plot.py

@@ -5,7 +5,6 @@ import matplotlib.patches as mpatches
 import matplotlib.pyplot as plt
 from matplotlib.axes import Axes
 
-from phylogenie import Tree
 from phylogenie.tree import Tree
 from phylogenie.utils import get_times
 
@@ -58,6 +57,10 @@ def plot_tree(
             norm = mcolors.Normalize(vmin=min(values), vmax=max(values))
             colormap = plt.get_cmap(cmap)
             colors = {node: colormap(norm(float(node.get(color_by)))) for node in tree}
+
+            sm = plt.cm.ScalarMappable(cmap=colormap, norm=norm)
+            fig = ax.get_figure()
+            fig.colorbar(sm, ax=ax, label=color_by)  # pyright: ignore
         else:
             raise ValueError(
                 f"Unknown coloring method: {coloring}. Choices are {list(Coloring)}."

{phylogenie-2.1.10 → phylogenie-2.1.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "phylogenie"
-version = "2.1.10"
+version = "2.1.12"
 description = "Generate phylogenetic datasets with minimal setup effort"
 authors = ["Gabriele Marino <gabmarino.8601@gmail.com>"]
 readme = "README.md"