Update infer.py

infer.py

@@ -16,8 +16,8 @@ from rdkit import Chem
 
 class ModelInference(object):
     def __init__(self, config_path, pretrain_model_path, device):
-        assert (config_path is not None, "config_path is None")
-        assert (pretrain_model_path is not None, "pretrain_model_path is None")
+        assert config_path is not None, "config_path is None"
+        assert pretrain_model_path is not None, "pretrain_model_path is None"
 
         if device is None:
             self.device = torch.device(