fix rna alphabet to avoid one-hot feature crash

This commit is contained in:
clami66 2024-06-12 14:31:13 +02:00
parent bf214835d6
commit a7afaf0b83

View file

@ -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: