Commit a2f30d65 authored by seyonechithrananda's avatar seyonechithrananda
Browse files

remove print with assert

parent bd3b8ac2
Loading
Loading
Loading
Loading
+0 −1
Original line number Diff line number Diff line
@@ -18,6 +18,5 @@ class TestSmilesTokenizer(TestCase):
      model.num_parameters()

      tokenizer = SmilesTokenizer(vocab_path, max_len=model.config.max_position_embeddings)
      print(tokenizer.encode("CCC(CC)COC(=O)[C@H](C)N[P@](=O)(OC[C@H]1O[C@](C#N)([C@H](O)[C@@H]1O)C1=CC=C2N1N=CN=C2N)OC1=CC=CC=C1"))

      assert tokenized_smiles == tokenizer.encode("CCC(CC)COC(=O)[C@H](C)N[P@](=O)(OC[C@H]1O[C@](C#N)([C@H](O)[C@@H]1O)C1=CC=C2N1N=CN=C2N)OC1=CC=CC=C1")