Added calls to prepare() to ensure requirement checks are done early

Signed-off-by: Yoav Katz <katz@il.ibm.com>

src/unitxt/formats.py

@@ -229,6 +229,7 @@ class HFSystemFormat(BaseFormat):
     _requirements_list = ["transformers"]
 
     def prepare(self):
+        super().prepare()
         from transformers import AutoTokenizer
 
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)

src/unitxt/image_operators.py

@@ -47,6 +47,9 @@ def process_instance_value(self, value: Any, instance: Dict[str, Any]):
 
 
 class ImageFieldOperator(FieldOperator, PillowMixin):
+    # Required here, because PillowMixin _requirements_list is not inherited
+    _requirements_list = {"PIL": "pip install pillow"}
+
     @abstractmethod
     def process_image(self, image):
         pass

src/unitxt/inference.py

@@ -30,6 +30,7 @@ def prepare_engine(self):
         pass
 
     def prepare(self):
+        super().prepare()
         if not settings.mock_inference_mode:
             self.prepare_engine()
 

src/unitxt/operator.py

@@ -30,8 +30,8 @@ class PackageRequirementsMixin(Artifact):
     )
 
     def prepare(self):
-        super().prepare()
         self.check_missing_requirements()
+        super().prepare()
 
     def check_missing_requirements(self, requirements=None):
         if requirements is None: