Commit 4cd8e6cb authored by seyonechithrananda's avatar seyonechithrananda
Browse files

add pytest import

parent 6d35dfa0
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -2,6 +2,7 @@
import os
import unittest
from unittest import TestCase
import pytest
try:
  from transformers import RobertaForMaskedLM
  from deepchem.feat.smiles_tokenizer import SmilesTokenizer