phylogenie 2.1.10__py3-none-any.whl → 2.1.12__py3-none-any.whl

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/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.dist-info → phylogenie-2.1.12.dist-info}/METADATA

@@ -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.dist-info → phylogenie-2.1.12.dist-info}/RECORD

@@ -1,4 +1,4 @@
-phylogenie/__init__.py,sha256=T2mRLsYtoLlWt8GlxrrUnfXJ9XVioq7hTvVq3uJpwQI,2215
+phylogenie/__init__.py,sha256=FQE_BakIftKaCnMaCZ4qpoSUH0JJPh2igAZLl_5pA28,2296
 phylogenie/generators/__init__.py,sha256=zsOxy28-9j9alOQLIgrOAFfmM58NNHO_NEtW-KXQXAY,888
 phylogenie/generators/alisim.py,sha256=G7p6tfcDWncg4xl3NhEXxgikcZDUi_RdE33zWn_CvzY,2704
 phylogenie/generators/configs.py,sha256=WFoeKpgj9ZQIom7BKqwpgXbriiQGg3jFBMLoD8KButk,1073
@@ -10,7 +10,7 @@ phylogenie/io.py,sha256=nwy8DOknt0HqF9qMeFZHrCmSXpM5AGrU5oajwTtD6vY,3973
 phylogenie/main.py,sha256=vtvSpQxBNlYABoFQ25czl-l3fIr4QRo3svWVd-jcArw,1170
 phylogenie/models.py,sha256=pCg9ob0RpLUHwM49x4knKxL4FNPr3-EU_6zMXsvxtAg,370
 phylogenie/msa.py,sha256=JDGyZUsAq6-m-SQjoCDjAkAZIxfgyl_PDIhdYn5HOow,2064
-phylogenie/plot.py,sha256=oQYlyKnahGSnC6pyWm0wHpGDzGorLjOJMCIms-sUXd0,2856
+phylogenie/plot.py,sha256=hjftVhapi9AoD2_e200E82rVtYHY5dNptvyMKYcbnQI,2999
 phylogenie/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phylogenie/skyline/__init__.py,sha256=7pF4CUb4ZCLzNYJNhOjpuTOLTRhlK7L6ugfccNqjIGo,620
 phylogenie/skyline/matrix.py,sha256=Gl8OgKjtieG0NwPYiPimKI36gefV8fm_OeorjdXxPTs,9146
@@ -28,8 +28,8 @@ phylogenie/treesimulator/model.py,sha256=Ct0lfn6maKtjuFxivWx1MbFHvH3Y-fiJ0XXMdkN
 phylogenie/typeguards.py,sha256=JtqmbEWJZBRHbWgCvcl6nrWm3VcBfzRbklbTBYHItn0,1325
 phylogenie/typings.py,sha256=GknvAFXyiaWeeYJ8Lk5d6E2VHT-xW6ONEojYbtJYiB8,476
 phylogenie/utils.py,sha256=tVorfhMuRkxogeNu45aBv4SDBflHg4faiIpnHNJPKPY,871
-phylogenie-2.1.10.dist-info/LICENSE.txt,sha256=NUrDqElK-eD3I0WqC004CJsy6cs0JgsAoebDv_42-pw,1071
-phylogenie-2.1.10.dist-info/METADATA,sha256=NRjmmmqUcJKklEr5G7mGPe_qrWv0Z7CQxvV1Mkz8sOc,5477
-phylogenie-2.1.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-phylogenie-2.1.10.dist-info/entry_points.txt,sha256=Rt6_usN0FkBX1ZfiqCirjMN9FKOgFLG8rydcQ8kugeE,51
-phylogenie-2.1.10.dist-info/RECORD,,
+phylogenie-2.1.12.dist-info/LICENSE.txt,sha256=NUrDqElK-eD3I0WqC004CJsy6cs0JgsAoebDv_42-pw,1071
+phylogenie-2.1.12.dist-info/METADATA,sha256=65IaQc4XK0qJH_CwrqkdHtVpU2u8mxQLZxuzpDIljK8,5477
+phylogenie-2.1.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+phylogenie-2.1.12.dist-info/entry_points.txt,sha256=Rt6_usN0FkBX1ZfiqCirjMN9FKOgFLG8rydcQ8kugeE,51
+phylogenie-2.1.12.dist-info/RECORD,,