bug fixes to adress #303

gtonkinhill · Sep 20, 2024 · 1662aed · 1662aed
1 parent e928a7a
commit 1662aed
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/panaroo/cdhit.py b/panaroo/cdhit.py
@@ -190,7 +190,7 @@ def cluster_nodes_cdhit(
                       s=s,
                       aL=aL,
                       AL=AL,
-                      aS=AS,
+                      aS=aS,
                       accurate=accurate,
                       use_local=use_local,
                       strand=strand,
@@ -203,7 +203,7 @@ def cluster_nodes_cdhit(
                   s=s,
                   aL=aL,
                   AL=AL,
-                  aS=AS,
+                  aS=aS,
                   accurate=accurate,
                   use_local=use_local,
                   quiet=quiet,

diff --git a/panaroo/generate_network.py b/panaroo/generate_network.py
@@ -225,4 +225,7 @@ def generate_network(cluster_file, data_file, prot_seq_file, all_dna=False):
                                    members=intbitset([genome_id]))
                 prev = current_cluster
 
+    if prev is not None: 
+        G.nodes[prev]['hasEnd'] = True
+
     return G, centroid_context, seqid_to_centroid