phylogenie 1.0.6__tar.gz → 1.0.7__tar.gz

{phylogenie-1.0.6 → phylogenie-1.0.7}/PKG-INFO

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

{phylogenie-1.0.6 → phylogenie-1.0.7}/phylogenie/backend/treesimulator.py

@@ -101,6 +101,7 @@ def generate_tree(
     )
     for i, leaf in enumerate(tree.iter_leaves()):
         state: str = getattr(leaf, STATE)
+        delattr(leaf, STATE)
         date = tree.get_distance(leaf)
         leaf.name = f"{i}|{state}|{date}"
     save_forest([tree], output_file)

{phylogenie-1.0.6 → phylogenie-1.0.7}/phylogenie/core/msas/base.py

@@ -1,6 +1,7 @@
 import os
 from abc import abstractmethod
 from enum import Enum
+from pathlib import Path
 from typing import Literal
 
 from numpy.random import Generator
@@ -14,9 +15,14 @@ class BackendType(str, Enum):
     ALISIM = "alisim"
 
 
+MSAS_DIRNAME = "MSAs"
+TREES_DIRNAME = "trees"
+
+
 class MSAsGenerator(DatasetGenerator):
     data_type: Literal[DataType.MSAS] = DataType.MSAS
     trees: TreesGeneratorConfig
+    keep_trees: bool = False
 
     @abstractmethod
     def _generate_one_from_tree(
@@ -24,11 +30,22 @@ class MSAsGenerator(DatasetGenerator):
     ) -> None: ...
 
     def _generate_one(self, filename: str, rng: Generator, data: pgt.Data) -> None:
-        tree_filename = f"{filename}.temp-tree"
+        if self.keep_trees:
+            base_dir = Path(filename).parent
+            file_id = Path(filename).stem
+            tree_filename = os.path.join(base_dir, TREES_DIRNAME, file_id)
+            msas_dir = os.path.join(base_dir, MSAS_DIRNAME)
+            os.makedirs(msas_dir, exist_ok=True)
+            msa_filename = os.path.join(msas_dir, file_id)
+        else:
+            tree_filename = f"{filename}.temp-tree"
+            msa_filename = filename
+
         self.trees.generate_one(
             filename=tree_filename, data=data, seed=int(rng.integers(0, 2**32 - 1))
         )
         self._generate_one_from_tree(
-            filename=filename, tree_file=f"{tree_filename}.nwk", rng=rng, data=data
+            filename=msa_filename, tree_file=f"{tree_filename}.nwk", rng=rng, data=data
         )
-        os.remove(f"{tree_filename}.nwk")
+        if not self.keep_trees:
+            os.remove(f"{tree_filename}.nwk")

{phylogenie-1.0.6 → phylogenie-1.0.7}/pyproject.toml

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