Commit 6d35dfa0 authored by seyonechithrananda's avatar seyonechithrananda
Browse files

add transformers req to torch

parent 5c8a9631
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -14,6 +14,7 @@ class TestSmilesTokenizer(TestCase):
  """Tests the SmilesTokenizer to load the USPTO vocab file and a ChemBERTa Masked LM model with pre-trained weights.."""

  @unittest.skipIf(not has_transformers, 'transformers are not installed')
  @pytest.mark.torch
  def test_tokenize(self):
    current_dir = os.path.dirname(os.path.realpath(__file__))
    vocab_path = os.path.join(current_dir, 'data', 'vocab.txt')
+1 −0
Original line number Diff line number Diff line
@@ -8,6 +8,7 @@ dependencies:
  - numpy==1.21.*
  - pip==20.2.*
  - pip:
    - transformers==4.6.*
    - tensorflow~=2.4
    - dgllife==0.2.8
    - pymatgen==2020.12.31