Transformers

molfeat/trans/pretrained/hf_transformers.py

@@ -13,20 +13,23 @@
 
 from dataclasses import dataclass
 from loguru import logger
-from transformers import EncoderDecoderModel
-from transformers import AutoTokenizer
-from transformers import AutoModel
-from transformers import AutoConfig
-from transformers import MODEL_MAPPING
-from transformers import PreTrainedModel
-from transformers import PreTrainedTokenizer, PreTrainedTokenizerFast
+from molfeat.utils import requires
 from molfeat.trans.pretrained.base import PretrainedMolTransformer
 from molfeat.store.loader import PretrainedStoreModel
 from molfeat.store import ModelStore
 from molfeat.store import ModelInfo
 from molfeat.utils.converters import SmilesConverter
 from molfeat.utils.pooler import get_default_hgf_pooler
 
+if requires.check("transformers"):
+    from transformers import EncoderDecoderModel
+    from transformers import AutoTokenizer
+    from transformers import AutoModel
+    from transformers import AutoConfig
+    from transformers import MODEL_MAPPING
+    from transformers import PreTrainedModel
+    from transformers import PreTrainedTokenizer, PreTrainedTokenizerFast
+
 
 @dataclass
 class HFExperiment:
@@ -260,6 +263,11 @@ def __init__(
             random_seed: random seed to use for reproducibility whenever a DNN pooler is used (e.g bert/roberta)
         """
 
+        if not requires.check("transformers"):
+            raise ValueError(
+                "Cannot find transformers and/or tokenizers. It's required for this featurizer !"
+            )
+
         super().__init__(
             dtype=dtype,
             device=device,

tests/test_pretrained.py

@@ -124,7 +124,9 @@ def test_dgl_pretrained_cache(self):
         # add buffers
         self.assertLessEqual(cached_run, ori_run + time_buffer)
 
-
+@pytest.mark.xfail(
+    not requires.check("transformers"), reason="3rd party module transformers is missing"
+)
 class TestHGFTransformer(ut.TestCase):
     r"""Test cases for FingerprintsTransformer"""
     smiles = [