change name to compute_loglikelihood

DeepChainBio · Apr 30, 2021 · f25adbb · f25adbb
1 parent d414e29
commit f25adbb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/biotransformers/tests/test_esm.py b/biotransformers/tests/test_esm.py
@@ -10,7 +10,7 @@
 
 def test_esm_loglikelihood():
     test_trans = BioTransformers("esm1_t34_670M_UR100")
-    loglikelihood = test_trans.compute_loglikelihoods(test_sequences)
+    loglikelihood = test_trans.compute_loglikelihood(test_sequences)
     assert isinstance(loglikelihood, np.ndarray)
 
 

diff --git a/biotransformers/tests/test_protbert.py b/biotransformers/tests/test_protbert.py
@@ -10,7 +10,7 @@
 
 def test_protbert_loglikelihood():
     test_trans = BioTransformers("protbert")
-    loglikelihood = test_trans.compute_loglikelihoods(test_sequences)
+    loglikelihood = test_trans.compute_loglikelihood(test_sequences)
     assert isinstance(loglikelihood, np.ndarray)
 
 

diff --git a/biotransformers/wrappers/transformers_wrappers.py b/biotransformers/wrappers/transformers_wrappers.py
@@ -486,7 +486,7 @@ def compute_logits(
 
         return logits.numpy(), labels.numpy()
 
-    def compute_loglikelihoods(
+    def compute_loglikelihood(
         self,
         sequences_list: List[str],
         batch_size: int = 1,