Apply linter

IBM · Feb 6, 2025 · a06d77e · a06d77e
1 parent bad3386
commit a06d77e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/unitxt/metrics.py b/src/unitxt/metrics.py
@@ -6171,10 +6171,10 @@ def compute(self, references: List[Any], prediction: Any, task_data: Dict) -> di
             self._tokenizer = AutoTokenizer.from_pretrained(self.hf_model_name)
             self.inference_engine = WMLInferenceEngineGeneration(
                 model_name=self.model_name,
-                decoding_method= "greedy",
-                max_new_tokens= 20,
-                temperature= 0,
-                return_options= {
+                decoding_method="greedy",
+                max_new_tokens=20,
+                temperature=0,
+                return_options={
                     "token_logprobs": True,
                     "generated_tokens": True,
                     "input_text": True,
@@ -6205,7 +6205,7 @@ def compute(self, references: List[Any], prediction: Any, task_data: Dict) -> di
         logger.debug(f"Results are ready:\n{result}")
         return result
 
-    def create_message(self, role: str, content: str) -> list[dict[str, str]]:
+    def create_message(self, role: str, content: str) -> List[Dict[str, str]]:
         return [{"role": role, "content": content}]
 
     def process_input_fields(self, task_data):