phylogenie 3.1.7__tar.gz → 3.1.11__tar.gz

{phylogenie-3.1.7/src/phylogenie.egg-info → phylogenie-3.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: phylogenie
-Version: 3.1.7
+Version: 3.1.11
 Summary: Generate phylogenetic datasets with minimal setup effort
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -23,7 +23,6 @@ Dynamic: license-file
 [![PyPI version](https://img.shields.io/pypi/v/phylogenie)](https://pypi.org/project/phylogenie/)
 ![Downloads](https://img.shields.io/pypi/dm/phylogenie)
 
-
 Phylogenie is a [Python](https://www.python.org/) package designed to easily simulate phylogenetic datasets—such as trees and multiple sequence alignments (MSAs)—with minimal setup effort. Simply specify the distributions from which your parameters should be sampled, and Phylogenie will handle the rest!
 
 ## ✨ Features
@@ -73,21 +72,21 @@ Phylogenie relies on [AliSim](https://iqtree.github.io/doc/AliSim) for simulatin
 
 ## 🚀 Quick Start
 
-Once you have installed Phylogenie, check out the [examples](https://github.com/gabriele-marino/phylogenie/tree/main/examples) folder.  
+Once you have installed Phylogenie, check out the [tutorials](https://github.com/gabriele-marino/phylogenie/tree/main/tutorials) folder.  
 It includes a collection of thoroughly commented configuration files, organized as a step-by-step tutorial. These examples will help you understand how to use Phylogenie in practice and can be easily adapted to fit your own workflow.
 
 For quick start, pick your favorite config file and run Phylogenie with:
 ```bash
-phylogenie examples/config_file.yaml
+phylogenie tutorials/config_file.yaml
 ```
 This command will create the output dataset in the folder specified inside the configuration file, including data directories and metadata files for each dataset split defined in the config.
 
 >❗ *Tip*: Can’t choose just one config file?
-You can run them all at once by pointing Phylogenie to the folder! Just use: `phylogenie examples`. In this mode, Phylogenie will automatically find all `.yaml` files in the folder you specified and run for each of them!
+You can run them all at once by pointing Phylogenie to the folder! Just use: `phylogenie tutorials`. In this mode, Phylogenie will automatically find all `.yaml` files in the folder you specified and run for each of them!
 
 ## 📖 Documentation
 
-- The [examples](https://github.com/gabriele-marino/phylogenie/tree/main/examples) folder contains many ready-to-use, extensively commented configuration files that serve as a step-by-step tutorial to guide you through using Phylogenie. You can explore them to learn how it works or adapt them directly to your own workflows.
+- The [tutorials](https://github.com/gabriele-marino/phylogenie/tree/main/tutorials) folder contains many ready-to-use, extensively commented configuration files that serve as a step-by-step tutorial to guide you through using Phylogenie. You can explore them to learn how it works or adapt them directly to your own workflows.
 - A complete user guide and API reference are under development. In the meantime, feel free to [reach out](mailto:gabmarino.8601@email.com) if you have any questions about integrating Phylogenie into your workflows.
 
 ## 📄 License

{phylogenie-3.1.7 → phylogenie-3.1.11}/README.md

@@ -8,7 +8,6 @@
 [![PyPI version](https://img.shields.io/pypi/v/phylogenie)](https://pypi.org/project/phylogenie/)
 ![Downloads](https://img.shields.io/pypi/dm/phylogenie)
 
-
 Phylogenie is a [Python](https://www.python.org/) package designed to easily simulate phylogenetic datasets—such as trees and multiple sequence alignments (MSAs)—with minimal setup effort. Simply specify the distributions from which your parameters should be sampled, and Phylogenie will handle the rest!
 
 ## ✨ Features
@@ -58,21 +57,21 @@ Phylogenie relies on [AliSim](https://iqtree.github.io/doc/AliSim) for simulatin
 
 ## 🚀 Quick Start
 
-Once you have installed Phylogenie, check out the [examples](https://github.com/gabriele-marino/phylogenie/tree/main/examples) folder.  
+Once you have installed Phylogenie, check out the [tutorials](https://github.com/gabriele-marino/phylogenie/tree/main/tutorials) folder.  
 It includes a collection of thoroughly commented configuration files, organized as a step-by-step tutorial. These examples will help you understand how to use Phylogenie in practice and can be easily adapted to fit your own workflow.
 
 For quick start, pick your favorite config file and run Phylogenie with:
 ```bash
-phylogenie examples/config_file.yaml
+phylogenie tutorials/config_file.yaml
 ```
 This command will create the output dataset in the folder specified inside the configuration file, including data directories and metadata files for each dataset split defined in the config.
 
 >❗ *Tip*: Can’t choose just one config file?
-You can run them all at once by pointing Phylogenie to the folder! Just use: `phylogenie examples`. In this mode, Phylogenie will automatically find all `.yaml` files in the folder you specified and run for each of them!
+You can run them all at once by pointing Phylogenie to the folder! Just use: `phylogenie tutorials`. In this mode, Phylogenie will automatically find all `.yaml` files in the folder you specified and run for each of them!
 
 ## 📖 Documentation
 
-- The [examples](https://github.com/gabriele-marino/phylogenie/tree/main/examples) folder contains many ready-to-use, extensively commented configuration files that serve as a step-by-step tutorial to guide you through using Phylogenie. You can explore them to learn how it works or adapt them directly to your own workflows.
+- The [tutorials](https://github.com/gabriele-marino/phylogenie/tree/main/tutorials) folder contains many ready-to-use, extensively commented configuration files that serve as a step-by-step tutorial to guide you through using Phylogenie. You can explore them to learn how it works or adapt them directly to your own workflows.
 - A complete user guide and API reference are under development. In the meantime, feel free to [reach out](mailto:gabmarino.8601@email.com) if you have any questions about integrating Phylogenie into your workflows.
 
 ## 📄 License

{phylogenie-3.1.7 → phylogenie-3.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "phylogenie"
-version = "3.1.7"
+version = "3.1.11"
 description = "Generate phylogenetic datasets with minimal setup effort"
 readme = "README.md"
 requires-python = ">=3.10"

{phylogenie-3.1.7 → phylogenie-3.1.11}/src/phylogenie/__init__.py

@@ -1,11 +1,3 @@
-from phylogenie.draw import (
-    draw_colored_dated_tree_categorical,
-    draw_colored_dated_tree_continuous,
-    draw_colored_tree_categorical,
-    draw_colored_tree_continuous,
-    draw_dated_tree,
-    draw_tree,
-)
 from phylogenie.generators import (
     AliSimDatasetGenerator,
     BDEITreeDatasetGenerator,
@@ -71,12 +63,6 @@ from phylogenie.treesimulator import (
 )
 
 __all__ = [
-    "draw_colored_dated_tree_categorical",
-    "draw_colored_dated_tree_continuous",
-    "draw_colored_tree_categorical",
-    "draw_colored_tree_continuous",
-    "draw_dated_tree",
-    "draw_tree",
     "AliSimDatasetGenerator",
     "BDEITreeDatasetGenerator",
     "BDSSTreeDatasetGenerator",

{phylogenie-3.1.7 → phylogenie-3.1.11}/src/phylogenie/draw.py

@@ -1,5 +1,5 @@
+import datetime
 from dataclasses import dataclass
-from datetime import datetime
 from typing import Any, Literal, overload
 
 import matplotlib.colors as mcolors
@@ -10,13 +10,18 @@ from matplotlib.axes import Axes
 from matplotlib.colors import Colormap
 from mpl_toolkits.axes_grid1.inset_locator import inset_axes  # pyright: ignore
 
-from phylogenie.treesimulator import Tree, get_node_depth_levels, get_node_depths
+from phylogenie.treesimulator import (
+    Tree,
+    get_node_ages,
+    get_node_depth_levels,
+    get_node_depths,
+)
 
 
 @dataclass
 class CalibrationNode:
     node: Tree
-    date: datetime
+    date: datetime.date
 
 
 Color = str | tuple[float, float, float] | tuple[float, float, float, float]
@@ -27,6 +32,8 @@ def draw_tree(
     ax: Axes | None = None,
     colors: Color | dict[Tree, Color] = "black",
     backward_time: bool = False,
+    branch_kwargs: dict[str, Any] | None = None,
+    sampled_ancestor_kwargs: dict[str, Any] | None = None,
 ) -> Axes:
     """
     Draw a phylogenetic tree with colored branches.
@@ -41,6 +48,10 @@ def draw_tree(
         A single color for all branches or a dictionary mapping each node to a color.
     backward_time : bool, optional
         If True, the x-axis is inverted to represent time going backward.
+    branch_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to pass to the branch drawing functions.
+    sampled_ancestor_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to highlight sampled ancestors.
 
     Returns
     -------
@@ -49,33 +60,51 @@ def draw_tree(
     """
     if ax is None:
         ax = plt.gca()
+    if branch_kwargs is None:
+        branch_kwargs = {}
+    if sampled_ancestor_kwargs is None:
+        sampled_ancestor_kwargs = {}
+    if "marker" not in sampled_ancestor_kwargs:
+        sampled_ancestor_kwargs["marker"] = "o"
 
     if not isinstance(colors, dict):
         colors = {node: colors for node in tree}
 
     xs = (
-        get_node_depth_levels(tree)
+        get_node_ages(tree)
+        if backward_time
+        else get_node_depth_levels(tree)
         if any(node.branch_length is None for node in tree.iter_descendants())
         else get_node_depths(tree)
     )
-    if backward_time:
-        max_x = max(xs.values())
-        xs = {node: max_x - x for node, x in xs.items()}
 
-    ys: dict[Tree, float] = {node: i for i, node in enumerate(tree.get_leaves())}
+    leaves = tree.get_leaves()
+    ys: dict[Tree, float] = {
+        node: i
+        for i, node in enumerate(leaves)
+        if node.parent is None or node.branch_length != 0
+    }
     for node in tree.postorder_traversal():
         if node.is_internal():
-            ys[node] = sum(ys[child] for child in node.children) / len(node.children)
+            children = [child for child in node.children if child.branch_length != 0]
+            ys[node] = sum(ys[child] for child in children) / len(children)
+    for leaf in leaves:
+        if leaf.parent is not None and leaf.branch_length == 0:
+            ys[leaf] = ys[leaf.parent]
 
     if tree.branch_length is not None:
         xmin = xs[tree] + tree.branch_length if backward_time else 0
-        ax.hlines(y=ys[tree], xmin=xmin, xmax=xs[tree], color=colors[tree])  # pyright: ignore
+        ax.hlines(  # pyright: ignore
+            y=ys[tree], xmin=xmin, xmax=xs[tree], color=colors[tree], **branch_kwargs
+        )
     for node in tree:
         x1, y1 = xs[node], ys[node]
+        if node.parent is not None and node.branch_length == 0:
+            ax.plot(x1, y1, color=colors[node], **sampled_ancestor_kwargs)  # pyright: ignore
         for child in node.children:
             x2, y2 = xs[child], ys[child]
-            ax.hlines(y=y2, xmin=x1, xmax=x2, color=colors[child])  # pyright: ignore
-            ax.vlines(x=x1, ymin=y1, ymax=y2, color=colors[child])  # pyright: ignore
+            ax.hlines(y=y2, xmin=x1, xmax=x2, color=colors[child], **branch_kwargs)  # pyright: ignore
+            ax.vlines(x=x1, ymin=y1, ymax=y2, color=colors[child], **branch_kwargs)  # pyright: ignore
 
     if backward_time:
         ax.invert_xaxis()
@@ -86,7 +115,7 @@ def draw_tree(
 
 def _depth_to_date(
     depth: float, calibration_nodes: tuple[CalibrationNode, CalibrationNode]
-) -> datetime:
+) -> datetime.date:
     """
     Convert a depth value to a date using linear interpolation between two calibration nodes.
 
@@ -99,7 +128,7 @@ def _depth_to_date(
 
     Returns
     -------
-    datetime
+    datetime.date
         The interpolated date corresponding to the given depth.
     """
     node1, node2 = calibration_nodes
@@ -113,6 +142,7 @@ def draw_dated_tree(
     calibration_nodes: tuple[CalibrationNode, CalibrationNode],
     ax: Axes | None = None,
     colors: Color | dict[Tree, Color] = "black",
+    branch_kwargs: dict[str, Any] | None = None,
 ) -> Axes:
     """
     Draw a phylogenetic tree with branches positioned according to calibrated dates.
@@ -127,6 +157,8 @@ def draw_dated_tree(
         The matplotlib Axes to draw on. If None, uses the current Axes.
     colors : Color | dict[Tree, Color], optional
         A single color for all branches or a dictionary mapping each node to a color.
+    branch_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to pass to the branch drawing functions.
 
     Returns
     -------
@@ -135,6 +167,8 @@ def draw_dated_tree(
     """
     if ax is None:
         ax = plt.gca()
+    if branch_kwargs is None:
+        branch_kwargs = {}
 
     if not isinstance(colors, dict):
         colors = {node: colors for node in tree}
@@ -156,6 +190,7 @@ def draw_dated_tree(
             xmin=mdates.date2num(origin_date),  # pyright: ignore
             xmax=mdates.date2num(xs[tree]),  # pyright: ignore
             color=colors[tree],
+            **branch_kwargs,
         )
     for node in tree:
         x1, y1 = xs[node], ys[node]
@@ -166,8 +201,15 @@ def draw_dated_tree(
                 xmin=mdates.date2num(x1),  # pyright: ignore
                 xmax=mdates.date2num(x2),  # pyright: ignore
                 color=colors[child],
+                **branch_kwargs,
+            )
+            ax.vlines(  # pyright: ignore
+                x=mdates.date2num(x1),  # pyright: ignore
+                ymin=y1,
+                ymax=y2,
+                color=colors[child],
+                **branch_kwargs,
             )
-            ax.vlines(x=mdates.date2num(x1), ymin=y1, ymax=y2, color=colors[child])  # pyright: ignore
 
     ax.xaxis.set_major_locator(mdates.AutoDateLocator())
     ax.xaxis.set_major_formatter(mdates.DateFormatter("%Y-%m-%d"))
@@ -256,6 +298,8 @@ def draw_colored_tree_categorical(
     show_legend: bool = True,
     labels: dict[Any, str] | None = None,
     legend_kwargs: dict[str, Any] | None = None,
+    branch_kwargs: dict[str, Any] | None = None,
+    sampled_ancestor_kwargs: dict[str, Any] | None = None,
 ):
     """
     Draw a phylogenetic tree with branches colored based on categorical metadata.
@@ -280,6 +324,10 @@ def draw_colored_tree_categorical(
         A mapping from category values to labels for the legend.
     legend_kwargs : dict[str, Any] | None, optional
         Additional keyword arguments to pass to the legend.
+    branch_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to pass to the branch drawing functions.
+    sampled_ancestor_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to highlight sampled ancestors.
 
     Returns
     -------
@@ -296,7 +344,14 @@ def draw_colored_tree_categorical(
         labels=labels,
         legend_kwargs=legend_kwargs,
     )
-    return draw_tree(tree=tree, ax=ax, colors=colors, backward_time=backward_time)
+    return draw_tree(
+        tree=tree,
+        ax=ax,
+        colors=colors,
+        backward_time=backward_time,
+        branch_kwargs=branch_kwargs,
+        sampled_ancestor_kwargs=sampled_ancestor_kwargs,
+    )
 
 
 def draw_colored_dated_tree_categorical(
@@ -309,6 +364,7 @@ def draw_colored_dated_tree_categorical(
     show_legend: bool = True,
     labels: dict[Any, str] | None = None,
     legend_kwargs: dict[str, Any] | None = None,
+    branch_kwargs: dict[str, Any] | None = None,
 ) -> Axes:
     """
     Draw a dated phylogenetic tree with branches colored based on categorical metadata.
@@ -333,6 +389,8 @@ def draw_colored_dated_tree_categorical(
         A mapping from category values to labels for the legend.
     legend_kwargs : dict[str, Any] | None, optional
         Additional keyword arguments to pass to the legend.
+    branch_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to pass to the branch drawing functions.
 
     Returns
     -------
@@ -350,7 +408,11 @@ def draw_colored_dated_tree_categorical(
         legend_kwargs=legend_kwargs,
     )
     return draw_dated_tree(
-        tree=tree, calibration_nodes=calibration_nodes, ax=ax, colors=colors
+        tree=tree,
+        calibration_nodes=calibration_nodes,
+        ax=ax,
+        colors=colors,
+        branch_kwargs=branch_kwargs,
     )
 
 
@@ -378,7 +440,7 @@ def _init_colored_tree_continuous(
     vmin: float | None = ...,
     vmax: float | None = ...,
     *,
-    show_hist: Literal[True],
+    show_hist: Literal[True] = True,
     hist_kwargs: dict[str, Any] | None = ...,
     hist_axes_kwargs: dict[str, Any] | None = ...,
 ) -> tuple[Axes, dict[Tree, Color], Axes]: ...
@@ -390,7 +452,6 @@ def _init_colored_tree_continuous(
     colormap: str | Colormap = "viridis",
     vmin: float | None = None,
     vmax: float | None = None,
-    *,
     show_hist: bool = True,
     hist_kwargs: dict[str, Any] | None = None,
     hist_axes_kwargs: dict[str, Any] | None = None,
@@ -474,6 +535,8 @@ def draw_colored_tree_continuous(
     colormap: str | Colormap = "viridis",
     vmin: float | None = None,
     vmax: float | None = None,
+    branch_kwargs: dict[str, Any] | None = None,
+    sampled_ancestor_kwargs: dict[str, Any] | None = None,
     *,
     show_hist: Literal[False],
     hist_kwargs: dict[str, Any] | None = None,
@@ -489,8 +552,9 @@ def draw_colored_tree_continuous(
     colormap: str | Colormap = "viridis",
     vmin: float | None = None,
     vmax: float | None = None,
-    *,
-    show_hist: Literal[True],
+    branch_kwargs: dict[str, Any] | None = None,
+    sampled_ancestor_kwargs: dict[str, Any] | None = None,
+    show_hist: Literal[True] = True,
     hist_kwargs: dict[str, Any] | None = None,
     hist_axes_kwargs: dict[str, Any] | None = None,
 ) -> tuple[Axes, Axes]: ...
@@ -503,6 +567,8 @@ def draw_colored_tree_continuous(
     colormap: str | Colormap = "viridis",
     vmin: float | None = None,
     vmax: float | None = None,
+    branch_kwargs: dict[str, Any] | None = None,
+    sampled_ancestor_kwargs: dict[str, Any] | None = None,
     show_hist: bool = True,
     hist_kwargs: dict[str, Any] | None = None,
     hist_axes_kwargs: dict[str, Any] | None = None,
@@ -528,6 +594,10 @@ def draw_colored_tree_continuous(
         The minimum value for normalization. If None, uses the minimum of the data.
     vmax : float | None, optional
         The maximum value for normalization. If None, uses the maximum of the data.
+    branch_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to pass to the branch drawing functions.
+    sampled_ancestor_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to highlight sampled ancestors.
     show_hist : bool, optional
         Whether to display a histogram of the continuous values.
     hist_kwargs : dict[str, Any] | None, optional
@@ -555,7 +625,12 @@ def draw_colored_tree_continuous(
             hist_axes_kwargs=hist_axes_kwargs,
         )
         return draw_tree(
-            tree=tree, ax=ax, colors=colors, backward_time=backward_time
+            tree=tree,
+            ax=ax,
+            colors=colors,
+            backward_time=backward_time,
+            branch_kwargs=branch_kwargs,
+            sampled_ancestor_kwargs=sampled_ancestor_kwargs,
         ), hist_ax
 
     ax, colors = _init_colored_tree_continuous(
@@ -570,7 +645,14 @@ def draw_colored_tree_continuous(
         hist_kwargs=hist_kwargs,
         hist_axes_kwargs=hist_axes_kwargs,
     )
-    return draw_tree(tree=tree, ax=ax, colors=colors, backward_time=backward_time)
+    return draw_tree(
+        tree=tree,
+        ax=ax,
+        colors=colors,
+        backward_time=backward_time,
+        branch_kwargs=branch_kwargs,
+        sampled_ancestor_kwargs=sampled_ancestor_kwargs,
+    )
 
 
 @overload
@@ -583,6 +665,7 @@ def draw_colored_dated_tree_continuous(
     colormap: str | Colormap = "viridis",
     vmin: float | None = None,
     vmax: float | None = None,
+    branch_kwargs: dict[str, Any] | None = None,
     *,
     show_hist: Literal[False],
     hist_kwargs: dict[str, Any] | None = None,
@@ -598,8 +681,8 @@ def draw_colored_dated_tree_continuous(
     colormap: str | Colormap = "viridis",
     vmin: float | None = None,
     vmax: float | None = None,
-    *,
-    show_hist: Literal[True],
+    branch_kwargs: dict[str, Any] | None = None,
+    show_hist: Literal[True] = True,
     hist_kwargs: dict[str, Any] | None = None,
     hist_axes_kwargs: dict[str, Any] | None = None,
 ) -> tuple[Axes, Axes]: ...
@@ -612,6 +695,7 @@ def draw_colored_dated_tree_continuous(
     colormap: str | Colormap = "viridis",
     vmin: float | None = None,
     vmax: float | None = None,
+    branch_kwargs: dict[str, Any] | None = None,
     show_hist: bool = True,
     hist_kwargs: dict[str, Any] | None = None,
     hist_axes_kwargs: dict[str, Any] | None = None,
@@ -637,6 +721,8 @@ def draw_colored_dated_tree_continuous(
         The minimum value for normalization. If None, uses the minimum of the data.
     vmax : float | None, optional
         The maximum value for normalization. If None, uses the maximum of the data.
+    branch_kwargs : dict[str, Any] | None, optional
+        Additional keyword arguments to pass to the branch drawing functions.
     show_hist : bool, optional
         Whether to display a histogram of the continuous values.
     hist_kwargs : dict[str, Any] | None, optional
@@ -668,6 +754,7 @@ def draw_colored_dated_tree_continuous(
             calibration_nodes=calibration_nodes,
             ax=ax,
             colors=colors,
+            branch_kwargs=branch_kwargs,
         ), hist_ax
 
     ax, colors = _init_colored_tree_continuous(
@@ -687,4 +774,5 @@ def draw_colored_dated_tree_continuous(
         calibration_nodes=calibration_nodes,
         ax=ax,
         colors=colors,
+        branch_kwargs=branch_kwargs,
     )

{phylogenie-3.1.7 → phylogenie-3.1.11}/src/phylogenie/io/fasta.py

@@ -1,3 +1,4 @@
+from datetime import date
 from pathlib import Path
 from typing import Callable
 
@@ -5,7 +6,8 @@ from phylogenie.msa import MSA, Sequence
 
 
 def load_fasta(
-    fasta_file: str | Path, extract_time_from_id: Callable[[str], float] | None = None
+    fasta_file: str | Path,
+    extract_time_from_id: Callable[[str], float | date] | None = None,
 ) -> MSA:
     sequences: list[Sequence] = []
     with open(fasta_file, "r") as f:
@@ -17,10 +19,14 @@ def load_fasta(
             if extract_time_from_id is not None:
                 time = extract_time_from_id(id)
             elif "|" in id:
+                last_metadata = id.split("|")[-1]
                 try:
-                    time = float(id.split("|")[-1])
+                    time = float(last_metadata)
                 except ValueError:
-                    pass
+                    try:
+                        time = date.fromisoformat(last_metadata)
+                    except ValueError:
+                        pass
             chars = next(f).strip()
             sequences.append(Sequence(id, chars, time))
     return MSA(sequences)

{phylogenie-3.1.7 → phylogenie-3.1.11}/src/phylogenie/msa.py

@@ -1,5 +1,6 @@
 from collections.abc import Iterator
 from dataclasses import dataclass
+from datetime import date
 
 import numpy as np
 
@@ -8,7 +9,7 @@ import numpy as np
 class Sequence:
     id: str
     chars: str
-    time: float | None = None
+    time: float | date | None = None
 
 
 class MSA:
@@ -25,8 +26,8 @@ class MSA:
         return [sequence.id for sequence in self.sequences]
 
     @property
-    def times(self) -> list[float]:
-        times: list[float] = []
+    def times(self) -> list[float | date]:
+        times: list[float | date] = []
         for sequence in self:
             if sequence.time is None:
                 raise ValueError(f"Time is not set for sequence {sequence.id}.")

{phylogenie-3.1.7 → phylogenie-3.1.11}/src/phylogenie/treesimulator/__init__.py

@@ -27,11 +27,14 @@ from phylogenie.treesimulator.utils import (
     compute_sackin_index,
     get_distance,
     get_mrca,
+    get_node_ages,
     get_node_depth_levels,
     get_node_depths,
     get_node_height_levels,
     get_node_heights,
     get_node_leaf_counts,
+    get_node_times,
+    get_path,
 )
 
 __all__ = [
@@ -65,9 +68,12 @@ __all__ = [
     "compute_sackin_index",
     "get_distance",
     "get_mrca",
+    "get_node_ages",
     "get_node_depth_levels",
     "get_node_depths",
     "get_node_height_levels",
     "get_node_heights",
     "get_node_leaf_counts",
+    "get_node_times",
+    "get_path",
 ]

{phylogenie-3.1.7 → phylogenie-3.1.11}/src/phylogenie/treesimulator/features.py

@@ -5,11 +5,13 @@ from phylogenie.treesimulator.events.mutations import get_mutation_id
 from phylogenie.treesimulator.model import get_node_state
 from phylogenie.treesimulator.tree import Tree
 from phylogenie.treesimulator.utils import (
+    get_node_ages,
     get_node_depth_levels,
     get_node_depths,
     get_node_height_levels,
     get_node_heights,
     get_node_leaf_counts,
+    get_node_times,
 )
 
 
@@ -22,6 +24,7 @@ def _get_mutations(tree: Tree) -> dict[Tree, int]:
 
 
 class Feature(str, Enum):
+    AGE = "age"
     DEPTH = "depth"
     DEPTH_LEVEL = "depth_level"
     HEIGHT = "height"
@@ -29,9 +32,11 @@ class Feature(str, Enum):
     MUTATION = "mutation"
     N_LEAVES = "n_leaves"
     STATE = "state"
+    TIME = "time"
 
 
 FEATURES_EXTRACTORS = {
+    Feature.AGE: get_node_ages,
     Feature.DEPTH: get_node_depths,
     Feature.DEPTH_LEVEL: get_node_depth_levels,
     Feature.HEIGHT: get_node_heights,
@@ -39,6 +44,7 @@ FEATURES_EXTRACTORS = {
     Feature.MUTATION: _get_mutations,
     Feature.N_LEAVES: get_node_leaf_counts,
     Feature.STATE: _get_states,
+    Feature.TIME: get_node_times,
 }
 
 

{phylogenie-3.1.7 → phylogenie-3.1.11}/src/phylogenie/treesimulator/tree.py

@@ -153,6 +153,16 @@ class Tree(MetadataMixin):
             child.branch_length_or_raise() + child.height for child in self.children
         )
 
+    @property
+    def time(self) -> float:
+        return self.depth
+
+    @property
+    def age(self) -> float:
+        if self.parent is None:
+            return self.height
+        return self.parent.age - self.branch_length_or_raise()
+
     # -------------
     # Miscellaneous
     # -------------

{phylogenie-3.1.7 → phylogenie-3.1.11}/src/phylogenie/treesimulator/utils.py

@@ -51,6 +51,17 @@ def get_node_heights(tree: Tree) -> dict[Tree, float]:
     return heights
 
 
+def get_node_times(tree: Tree) -> dict[Tree, float]:
+    return get_node_depths(tree)
+
+
+def get_node_ages(tree: Tree) -> dict[Tree, float]:
+    ages: dict[Tree, float] = {tree: tree.height}
+    for node in tree.iter_descendants():
+        ages[node] = ages[node.parent] - node.branch_length  # pyright: ignore
+    return ages
+
+
 def get_mrca(node1: Tree, node2: Tree) -> Tree:
     node1_ancestors = set(node1.iter_upward())
     for node2_ancestor in node2.iter_upward():

{phylogenie-3.1.7 → phylogenie-3.1.11/src/phylogenie.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: phylogenie
-Version: 3.1.7
+Version: 3.1.11
 Summary: Generate phylogenetic datasets with minimal setup effort
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -23,7 +23,6 @@ Dynamic: license-file
 [![PyPI version](https://img.shields.io/pypi/v/phylogenie)](https://pypi.org/project/phylogenie/)
 ![Downloads](https://img.shields.io/pypi/dm/phylogenie)
 
-
 Phylogenie is a [Python](https://www.python.org/) package designed to easily simulate phylogenetic datasets—such as trees and multiple sequence alignments (MSAs)—with minimal setup effort. Simply specify the distributions from which your parameters should be sampled, and Phylogenie will handle the rest!
 
 ## ✨ Features
@@ -73,21 +72,21 @@ Phylogenie relies on [AliSim](https://iqtree.github.io/doc/AliSim) for simulatin
 
 ## 🚀 Quick Start
 
-Once you have installed Phylogenie, check out the [examples](https://github.com/gabriele-marino/phylogenie/tree/main/examples) folder.  
+Once you have installed Phylogenie, check out the [tutorials](https://github.com/gabriele-marino/phylogenie/tree/main/tutorials) folder.  
 It includes a collection of thoroughly commented configuration files, organized as a step-by-step tutorial. These examples will help you understand how to use Phylogenie in practice and can be easily adapted to fit your own workflow.
 
 For quick start, pick your favorite config file and run Phylogenie with:
 ```bash
-phylogenie examples/config_file.yaml
+phylogenie tutorials/config_file.yaml
 ```
 This command will create the output dataset in the folder specified inside the configuration file, including data directories and metadata files for each dataset split defined in the config.
 
 >❗ *Tip*: Can’t choose just one config file?
-You can run them all at once by pointing Phylogenie to the folder! Just use: `phylogenie examples`. In this mode, Phylogenie will automatically find all `.yaml` files in the folder you specified and run for each of them!
+You can run them all at once by pointing Phylogenie to the folder! Just use: `phylogenie tutorials`. In this mode, Phylogenie will automatically find all `.yaml` files in the folder you specified and run for each of them!
 
 ## 📖 Documentation
 
-- The [examples](https://github.com/gabriele-marino/phylogenie/tree/main/examples) folder contains many ready-to-use, extensively commented configuration files that serve as a step-by-step tutorial to guide you through using Phylogenie. You can explore them to learn how it works or adapt them directly to your own workflows.
+- The [tutorials](https://github.com/gabriele-marino/phylogenie/tree/main/tutorials) folder contains many ready-to-use, extensively commented configuration files that serve as a step-by-step tutorial to guide you through using Phylogenie. You can explore them to learn how it works or adapt them directly to your own workflows.
 - A complete user guide and API reference are under development. In the meantime, feel free to [reach out](mailto:gabmarino.8601@email.com) if you have any questions about integrating Phylogenie into your workflows.
 
 ## 📄 License