Improve style.

Adibvafa · Sep 20, 2024 · 3186250 · 3186250
1 parent 38866f8
commit 3186250
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/CodonTransformer/CodonData.py b/CodonTransformer/CodonData.py
@@ -17,14 +17,14 @@
 from tqdm import tqdm
 
 from CodonTransformer.CodonUtils import (
-    STOP_SYMBOLS,
     AMBIGUOUS_AMINOACID_MAP,
     AMINO2CODON_TYPE,
     AMINO_ACIDS,
     ORGANISM2ID,
     START_CODONS,
     STOP_CODONS,
     STOP_SYMBOL,
+    STOP_SYMBOLS,
     find_pattern_in_fasta,
     get_taxonomy_id,
     sort_amino2codon_skeleton,

diff --git a/tests/test_CodonPrediction.py b/tests/test_CodonPrediction.py
@@ -1,20 +1,20 @@
+import random
 import unittest
 import warnings
-import random
 
 import torch
 
 from CodonTransformer.CodonData import get_amino_acid_sequence
-from CodonTransformer.CodonUtils import (
-    AMINO_ACIDS,
-    STOP_SYMBOLS,
-    ORGANISM2ID,
-)
 from CodonTransformer.CodonPrediction import (
     load_model,
     load_tokenizer,
     predict_dna_sequence,
 )
+from CodonTransformer.CodonUtils import (
+    AMINO_ACIDS,
+    ORGANISM2ID,
+    STOP_SYMBOLS,
+)
 
 
 class TestCodonPrediction(unittest.TestCase):