add timing

logscoremetric.py

@@ -19,6 +19,8 @@ import re
 import dateutil.parser
 import numpy as np
 
+import time
+
 
 # TODO: Add BibTeX citation
 _CITATION = """\
@@ -136,6 +138,8 @@ class LogScoreMetric(evaluate.Metric):
     def _compute(self, predictions, references):
         """Returns the scores"""
 
+        t_before = time.perf_counter()
+
         timestamp_score = np.mean([self.getLogMetric(p,r) for p,r in zip(predictions,references)])
         predictions_without_timestamps = [re.sub(self.timestamp_regex, '', p, flags=re.MULTILINE) for p in predictions]
         references_without_timestamps = [re.sub(self.timestamp_regex, '', r, flags=re.MULTILINE) for r in references]
@@ -143,7 +147,11 @@ class LogScoreMetric(evaluate.Metric):
         # Sacrebleu score on logs without timestamps
         sb_results = self.sacrebleu.compute(predictions=predictions_without_timestamps, references=references_without_timestamps)
 
+        t_after = time.perf_counter()
+        compute_duration = f" {t_after - t_before:0.4f}"
+
         return {
             "timestamp_score": timestamp_score,
             "sacrebleu_score": sb_results["score"],
+            "compute_duration":compute_duration
         }