Update cas9att.py (#11)

- Update cas9att.py (8227c5421df0ce2cd0f679aa6ac2ce140c145990)


Co-authored-by: Qingyang Liu <LfOreVEr@users.noreply.huggingface.co>

cas9att.py

@@ -224,8 +224,13 @@ def process_gene(gene_symbol, model_path):
     else:
         print("Failed to retrieve transcripts.")
 
+    output = []
+    for result in results:
+        for item in result:
+            output.append(item)
+    
     # Sort results based on prediction score (assuming score is at the 8th index)
-    sorted_results = sorted(results, key=lambda x: x[8], reverse=True)
+    sorted_results = sorted(output, key=lambda x: x[8], reverse=True)
 
     # Return the sorted output, combined gene sequences, and all exons
     return sorted_results, all_gene_sequences, all_exons