fix rna alphabet to avoid one-hot feature crash

baker-laboratory · Jun 12, 2024 · a7afaf0 · a7afaf0
1 parent bf21483
commit a7afaf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rf2aa/data/parsers.py b/rf2aa/data/parsers.py
@@ -227,7 +227,7 @@ def parse_multichain_fasta(filename,  maxseq=10000, rna_alphabet=False, dna_alph
 
     # convert letters into numbers
     if rna_alphabet:
-        alphabet = np.array(list("00000000000000000000-000000ACGTN"), dtype='|S1').view(np.uint8)
+        alphabet = np.array(list("00000000000000000000-000000ACGUN"), dtype='|S1').view(np.uint8)
     elif dna_alphabet:
         alphabet = np.array(list("00000000000000000000-0ACGTD00000"), dtype='|S1').view(np.uint8)
     else: