PyPI - LZGraphs - Versions diffs - 2.3.0__tar.gz → 2.3.1__tar.gz - Mend

@@ -1,5 +1,7 @@
 import copy
+import networkx as nx
 from ..graphs.edge_data import EdgeData
@@ -47,9 +49,17 @@ class BayesianPosteriorMixin:
         seq_list, abd_list = self._parse_posterior_input(sequences, abundances)
         ind_counts = self._extract_counts_for_posterior(seq_list, abd_list)
-        # 2. Create a lightweight copy (deep-copy only the nx.DiGraph)
+        # 2. Create a lightweight copy with a fast graph clone
+        #    (avoids copy.deepcopy which does 51M recursive calls on large graphs)
         posterior = copy.copy(self)
-        posterior.graph = copy.deepcopy(self.graph)
+        g = self.graph
+        new_g = nx.DiGraph()
+        new_g.add_nodes_from(g.nodes())
+        new_g.add_edges_from(
+            (a, b, {'data': g[a][b]['data'].copy()})
+            for a, b in g.edges()
+        )
+        posterior.graph = new_g
         # Deep copy mutable dicts that will be modified
         posterior.initial_state_counts = dict(self.initial_state_counts)

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LZGraphs
-Version: 2.3.0
+Version: 2.3.1
 Summary: An Implementation of LZ76 Based Graphs for Repertoire Representation and Analysis
 Author-email: Thomas Konstantinovsky <thomaskon90@gmail.com>
 Maintainer-email: Thomas Konstantinovsky <thomaskon90@gmail.com>

@@ -1,4 +1,4 @@
-__version__ = "2.3.0"
+__version__ = "2.3.1"
 # =============================================================================
 # Graph classes

@@ -87,6 +87,17 @@ class EdgeData:
             if self.j_genes[j_gene] <= 0:
                 del self.j_genes[j_gene]
+    def copy(self):
+        """Create a shallow copy of this EdgeData (fast, no pickle)."""
+        new = EdgeData.__new__(EdgeData)
+        new.count = self.count
+        new._weight = self._weight
+        new.v_genes = dict(self.v_genes)
+        new.j_genes = dict(self.j_genes)
+        new._vsum = self._vsum
+        new._jsum = self._jsum
+        return new
     def merge(self, other):
         """Merge another EdgeData into this one (for graph union).

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LZGraphs
-Version: 2.3.0
+Version: 2.3.1
 Summary: An Implementation of LZ76 Based Graphs for Repertoire Representation and Analysis
 Author-email: Thomas Konstantinovsky <thomaskon90@gmail.com>
 Maintainer-email: Thomas Konstantinovsky <thomaskon90@gmail.com>

LZGraphs 2.3.0__tar.gz → 2.3.1__tar.gz