Commit 5c8a9631 authored by seyonechithrananda's avatar seyonechithrananda
Browse files

add yapf formatting

parent a3a1f8ff
Loading
Loading
Loading
Loading
+6 −5
Original line number Diff line number Diff line
@@ -22,7 +22,8 @@ class TestSmilesTokenizer(TestCase):
        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]
        16, 16, 22, 16, 20, 13
    ]

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