Merge branch 'patch-4' of github.com:niklashoelter/deepchem into st_fix (a3a1f8ff) · Commits · 钟慕尧 / deepchem

deepchem/feat/tests/test_smiles_tokenizer.py

+5 −6

Original line number	Diff line number	Diff line
		@@ -19,11 +19,10 @@ class TestSmilesTokenizer(TestCase):
		vocab_path = os.path.join(current_dir, 'data', 'vocab.txt')
		tokenized_smiles = [
		12, 16, 16, 16, 17, 16, 16, 18, 16, 19, 16, 17, 22, 19, 18, 33, 17, 16,
		18, 23, 181, 17, 22, 19, 18, 17, 19, 16, 33, 20, 19, 55, 17, 16, 23, 18,
		17, 33, 17, 19, 18, 35, 20, 19, 18, 16, 20, 22, 16, 16, 22, 16, 21, 23,
		20, 23, 22, 16, 23, 22, 16, 21, 23, 18, 19, 16, 20, 22, 16, 16, 22, 16,
		16, 22, 16, 20, 13
		]
		18, 23, 181, 17, 22, 19, 18, 17, 19, 16, 33, 20, 19, 55, 17, 16, 38, 23,
		18, 17, 33, 17, 19, 18, 35, 20, 19, 18, 16, 20, 22, 16, 16, 22, 16, 21,
		23, 20, 23, 22, 16, 23, 22, 16, 21, 23, 18, 19, 16, 20, 22, 16, 16, 22,
		16, 16, 22, 16, 20, 13]

		model = RobertaForMaskedLM.from_pretrained(
		'seyonec/SMILES_tokenized_PubChem_shard00_50k')