Merge branch 'develop'

DeepChainBio · Jul 19, 2021 · 5ce7b19 · 5ce7b19
2 parents 740f5c6 + 316b577
commit 5ce7b19
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 6 deletions.
diff --git a/biotransformers/tests/test_loglikelihoods.py b/biotransformers/tests/test_loglikelihoods.py
@@ -24,8 +24,9 @@ def test_loglikelihoods_type_shape_and_range(
         normalize=True,
     )
     assert len(loglikelihoods) == len(sequences)
-    results = loglikelihoods_results[test_trans._model_dir][params]
-    assert_allclose(loglikelihoods, results, rtol=0.01)
+    if test_trans._model_dir in loglikelihoods_results.keys():
+        results = loglikelihoods_results[test_trans._model_dir][params]
+        assert_allclose(loglikelihoods, results, rtol=0.01)
 
 
 @pytest.mark.parametrize("batch_size, tokens_list, pass_mode", test_params_fasta)
@@ -47,5 +48,6 @@ def test_loglikelihoods_type_shape_and_range_fasta(
         normalize=True,
     )
     assert len(loglikelihoods) == len(lengths_sequence_fasta)
-    results = loglikelihoods_fasta_results[test_trans._model_dir]
-    assert_allclose(loglikelihoods, results, rtol=0.01)
+    if test_trans._model_dir in loglikelihoods_fasta_results.keys():
+        results = loglikelihoods_fasta_results[test_trans._model_dir]
+        assert_allclose(loglikelihoods, results, rtol=0.01)
diff --git a/biotransformers/utils/constant.py b/biotransformers/utils/constant.py
@@ -6,6 +6,7 @@
     "esm1_t6_43M_UR50S",
     "esm1b_t33_650M_UR50S",
     "esm_msa1_t12_100M_UR50S",
+    "esm1v_t33_650M_UR90S_1"
 ]
 
 ROSTLAB_LIST = ["Rostlab/prot_bert", "Rostlab/prot_bert_bfd"]

diff --git a/biotransformers/version.py b/biotransformers/version.py
@@ -1 +1 @@
-VERSION = "0.1.6"
+VERSION = "0.1.7"
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 biopython>=1.78,<2
-fair-esm==0.3.1
+fair-esm==0.4.0
 numpy>=1.16,<1.19
 pandas>=1.2.3
 pytorch_lightning>=1.3.3