release v0.1.5

Author: Luciennnnnnn

editscore/__init__.py

@@ -99,8 +99,8 @@ def evaluate(self, image_prompts, text_prompt):
                 if result_SC in ["I'm sorry, but I can't assist with that request."] or result_PQ in ["I'm sorry, but I can't assist with that request."]:
                     guess_if_cannot_parse = True
 
-                SC_dict = mllm_output_to_dict(result_SC, give_up_parsing=guess_if_cannot_parse, text_prompt=text_prompt)
-                PQ_dict = mllm_output_to_dict(result_PQ, give_up_parsing=guess_if_cannot_parse, text_prompt=text_prompt)
+                SC_dict = mllm_output_to_dict(result_SC, give_up_parsing=guess_if_cannot_parse, text_prompt=text_prompt, score_range=self.score_range)
+                PQ_dict = mllm_output_to_dict(result_PQ, give_up_parsing=guess_if_cannot_parse, text_prompt=text_prompt, score_range=self.score_range)
 
             if SC_dict == "rate_limit_exceeded" or PQ_dict == "rate_limit_exceeded":
                 print("rate_limit_exceeded") 

editscore/utils.py

@@ -201,7 +201,7 @@ def normalize_quotes(s: str) -> str:
     return s.replace("“", '"').replace("”", '"').replace("‘", "'").replace("’", "'")
 
 #+=========================================================================================
-def mllm_output_to_dict(input_string, give_up_parsing=False, text_prompt=None):
+def mllm_output_to_dict(input_string, give_up_parsing=False, text_prompt=None, score_range: int = 10):
     """
     Args:
         input_string (str): actually the output of the mllm model to be parsed
@@ -212,7 +212,7 @@ def mllm_output_to_dict(input_string, give_up_parsing=False, text_prompt=None):
         return "rate_limit_exceeded"
 
     if give_up_parsing:
-        guessed_value = random.randint(0, 10)
+        guessed_value = random.randint(0, score_range)
         json_content = {'score': [guessed_value], "reasoning": f"guess_if_cannot_parse | {input_string}"}
         return json_content
 

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "editscore"
-version = "0.1.4"
+version = "0.1.5"
 authors = [
   { name="Xin Luo", email="[email protected]" },
   { name="Jiahao Wang", email="[email protected]" },