minor change

jalammar · Jan 30, 2024 · 4bd5128 · 4bd5128
1 parent 0b27bf9
commit 4bd5128
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/lm_test.py b/tests/lm_test.py
@@ -1,12 +1,9 @@
-import numpy as np
+from ecco.lm import LM, _one_hot, sample_output_token, activations_dict_to_array
+import ecco
 import torch
+import numpy as np
 from transformers import PreTrainedModel
 
-import ecco
-from ecco.lm import _one_hot
-from ecco.lm import activations_dict_to_array
-from ecco.lm import sample_output_token
-
 
 class TestLM:
     def test_one_hot(self):