fix lint errors

allenai · Nov 9, 2024 · b7782d0 · b7782d0
1 parent 388927c
commit b7782d0
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 12 deletions.
diff --git a/contexteval/main/compute_human_agreement.py b/contexteval/main/compute_human_agreement.py
@@ -16,7 +16,6 @@
 import sys
 sys.path.append('contexteval/common')
 from scipy.stats import sem, t, ttest_ind, ttest_rel
-from statsmodels.stats import inter_rater as irr
 
 
 _EVAL_JUDGMENTS = flags.DEFINE_string(

diff --git a/contexteval/main/generate_analysis_evals.py b/contexteval/main/generate_analysis_evals.py
@@ -50,13 +50,8 @@ def get_qa(input_str):
 
 def get_eval_judgement(cur_prompt, model):
   dict_out = None
-  count = 0
   while dict_out is None:
     eval_judgement = model.generate(cur_prompt)
-    # dict_out = extract_dictionary(eval_judgement)
-    # count += 1
-    # if count == 3:
-    #   break
     break
   return eval_judgement
 

diff --git a/contexteval/main/generate_analysis_responses.py b/contexteval/main/generate_analysis_responses.py
@@ -44,9 +44,7 @@ def extract_dictionary(input_string):
     dict_str = input_string[dict_start:dict_end]
     dictionary = ast.literal_eval(dict_str)
     return dictionary
-  except Exception as e:
-    # print(input_string)
-    # print(f"Error occurred: {e}")
+  except Exception:
     return None
 
 

diff --git a/contexteval/main/generate_context_validation.py b/contexteval/main/generate_context_validation.py
@@ -48,10 +48,10 @@ def extract_dictionary(input_string):
     dict_str = input_string[dict_start:dict_end]
     dictionary = ast.literal_eval(dict_str)
     return dictionary
-  except Exception as e:
+  except Exception:
     return None
 
-  
+
 def check_output_format(input_string):
   try:
     input_string = input_string.replace("*", "").strip()
@@ -130,7 +130,7 @@ def main(unused_argv) -> None:
     tries = 0
     ex = None
     # Pick context generated by one of the three models
-    while skip==True:
+    while skip:
       if model_idx == 1:
         ex = examples[MODEL_ONE][idx]
       elif model_idx == 2: