LineageTree 1.6.0__tar.gz → 1.6.1__tar.gz

{lineagetree-1.6.0/src/LineageTree.egg-info → lineagetree-1.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: LineageTree
-Version: 1.6.0
+Version: 1.6.1
 Summary: Lineage tree structure
 Home-page: https://github.com/leoguignard/LineageTree
 Author: Léo Guignard

{lineagetree-1.6.0 → lineagetree-1.6.1}/pyproject.toml

@@ -10,7 +10,7 @@ profile = "black"
 line_length = 79
 
 [tool.bumpver]
-current_version = "1.6.0"
+current_version = "1.6.1"
 version_pattern = "MAJOR.MINOR.PATCH[-TAG]"
 commit_message = "bump version {old_version} -> {new_version}"
 commit = true

{lineagetree-1.6.0 → lineagetree-1.6.1}/src/LineageTree/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.6.0"
+__version__ = "1.6.1"
 from .lineageTree import lineageTree
 from .lineageTreeManager import lineageTreeManager
 

{lineagetree-1.6.0 → lineagetree-1.6.1}/src/LineageTree/lineageTree.py

@@ -1775,16 +1775,17 @@ class lineageTree(lineageTreeLoaders):
         selected_cells=None,
         color="magenta",
         ax=None,
+        figure=None,
         **kwargs,
     ):
         if selected_cells is None:
             selected_cells = []
         if ax is None:
-            _, ax = plt.subplots()
+            figure, ax = plt.subplots()
         else:
             ax.clear()
-
-        selected_cells = set(selected_cells)
+        if not isinstance(selected_cells, set):
+            selected_cells = set(selected_cells)
         hier_unselected = {
             k: v for k, v in hier.items() if k not in selected_cells
         }
@@ -1826,7 +1827,7 @@ class lineageTree(lineageTreeLoaders):
             )
         ax.get_yaxis().set_visible(False)
         ax.get_xaxis().set_visible(False)
-        return ax
+        return figure, ax
 
     def to_simple_graph(self, node=None, start_time: int = None):
         """Generates a dictionary of graphs where the keys are the index of the graph and

{lineagetree-1.6.0 → lineagetree-1.6.1/src/LineageTree.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: LineageTree
-Version: 1.6.0
+Version: 1.6.1
 Summary: Lineage tree structure
 Home-page: https://github.com/leoguignard/LineageTree
 Author: Léo Guignard