55from django .db import connection
66from django .utils .module_loading import import_string
77
8- from evaluate_m2 .evaluate_utils import create_eval_insert_query
9- from evaluate_m2 .upload_utils import stream_results_files_to_s3
10-
8+ from evaluate_m2 .evaluate_utils import create_eval_insert_query
119from evaluate_m2 .models import EvaluatorMetadata , EvaluatorResultSummary
12-
10+ from evaluate_m2 . upload_utils import stream_results_files_to_s3
1311from parse_m2 .models import Metro2Event
1412
1513
16- class Evaluate () :
14+ class Evaluate :
1715 # Evaluator version is saved on each evaluator result summary.
1816 # Increment this version for all updates to evaluator functionality.
1917 evaluator_version = "1.3"
@@ -27,12 +25,12 @@ def load_evaluators(self):
2725 for eval_id , eval_import_str in evaluators_dict .items ():
2826 try :
2927 eval_callable = import_string (eval_import_str )
30- except ImportError :
28+ except ImportError as e :
3129 raise ImproperlyConfigured (
3230 f"Unable to import { eval_import_str } for evaluator "
3331 f"{ eval_id } . Are you sure the package and evaluator "
3432 "callable exist?"
35- )
33+ ) from e
3634 self .evaluators [eval_id ] = eval_callable
3735
3836 # runs evaluators to produce results
@@ -82,7 +80,9 @@ def save_evaluator_results(self, result_summary, eval_query):
8280 with connection .cursor () as cursor :
8381 cursor .execute (full_query , query_params )
8482
85- def prepare_result_summary (self , event : Metro2Event , eval_id : str ) -> EvaluatorResultSummary :
83+ def prepare_result_summary (
84+ self , event : Metro2Event , eval_id : str
85+ ) -> EvaluatorResultSummary :
8686 """
8787 Create an EvaluatorResultSummary object so we can associate results with it.
8888 Later, we will update the values related to eval hits.
@@ -102,7 +102,9 @@ def prepare_result_summary(self, event: Metro2Event, eval_id: str) -> EvaluatorR
102102 evaluator_version = self .evaluator_version ,
103103 )
104104
105- def update_result_summary_with_actual_results (self , result_summary : EvaluatorResultSummary ):
105+ def update_result_summary_with_actual_results (
106+ self , result_summary : EvaluatorResultSummary
107+ ):
106108 """
107109 If the evaluator had any hits, update the information about the hits
108110 in the EvaluatorResultSummary record.
0 commit comments