LineageTree 1.6.0__py3-none-any.whl → 1.6.1__py3-none-any.whl

LineageTree/__init__.py

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

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.dist-info → LineageTree-1.6.1.dist-info}/METADATA

@@ -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.1.dist-info/RECORD

@@ -0,0 +1,11 @@
+LineageTree/__init__.py,sha256=fIOwpoeHI19rwQShQ50iRXyRg4OXzVo_A4q_BUDBmsk,155
+LineageTree/lineageTree.py,sha256=lamlTaHtu3gPUCMKIvlCHQqGMbcQAHCvgS8MB6EHWTQ,102742
+LineageTree/lineageTreeManager.py,sha256=tQMNmxAvUEd4fNLJViB-9DGItpgCRnSF-bS2_IdebMo,5566
+LineageTree/loaders.py,sha256=hKd-69d_csjcsV0zabSTEOyy0NhprFqvnRAI18tYa9w,23718
+LineageTree/tree_styles.py,sha256=IUEq8RHfwXTv3WXylpLVa5XoThaWt1i1cRsgZGXdd50,10634
+LineageTree/utils.py,sha256=kC90fi8TFm8Pz7oom_bDsq7HPPKnFdJ6wI-x36fLvHE,4962
+LineageTree-1.6.1.dist-info/LICENSE,sha256=IKncNCNpq93Kq7Ywg1V4I9Bu_99VMK-mX1EJyjTKLyc,1068
+LineageTree-1.6.1.dist-info/METADATA,sha256=eDkg_GjSQtFDXXjmyagn-gXRosFsDq2n8706srcixA8,4195
+LineageTree-1.6.1.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+LineageTree-1.6.1.dist-info/top_level.txt,sha256=CCqPCTX76zPTEUagUgTWbLaZun8752n59iOOwfUlvxs,12
+LineageTree-1.6.1.dist-info/RECORD,,

LineageTree-1.6.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-LineageTree/__init__.py,sha256=P4WGjjokRD5ugh27WC263Pwll1PmMdLgHyshMVLNvIc,155
-LineageTree/lineageTree.py,sha256=srhZ8UI4iBsKmSEIVhn0OpKuTSyrE6npEMxjj4v666M,102657
-LineageTree/lineageTreeManager.py,sha256=tQMNmxAvUEd4fNLJViB-9DGItpgCRnSF-bS2_IdebMo,5566
-LineageTree/loaders.py,sha256=hKd-69d_csjcsV0zabSTEOyy0NhprFqvnRAI18tYa9w,23718
-LineageTree/tree_styles.py,sha256=IUEq8RHfwXTv3WXylpLVa5XoThaWt1i1cRsgZGXdd50,10634
-LineageTree/utils.py,sha256=kC90fi8TFm8Pz7oom_bDsq7HPPKnFdJ6wI-x36fLvHE,4962
-LineageTree-1.6.0.dist-info/LICENSE,sha256=IKncNCNpq93Kq7Ywg1V4I9Bu_99VMK-mX1EJyjTKLyc,1068
-LineageTree-1.6.0.dist-info/METADATA,sha256=XlCCU5B8WqbKT0EiGqgf-kvVFH5D2aSWQzygUCc3YT4,4195
-LineageTree-1.6.0.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-LineageTree-1.6.0.dist-info/top_level.txt,sha256=CCqPCTX76zPTEUagUgTWbLaZun8752n59iOOwfUlvxs,12
-LineageTree-1.6.0.dist-info/RECORD,,