Skip to content

Commit 69298b4

Browse files
authored
fix older pyterrier references (#47)
1 parent 8cfcb7b commit 69298b4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onir_pt/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def __init__(self, pt_transformer, va_run, va_qrels, config):
3535
def validate(self):
3636
with _logger.duration('validation'):
3737
res = self.pt_transformer.transform(self.va_run)
38-
metrics = pyterrier.Utils.evaluate(res, self.va_qrels, metrics=self.config['va_metrics'])
38+
metrics = pyterrier.Evaluate(res, self.va_qrels, metrics=self.config['va_metrics'])
3939
primary_metric = metrics[self.config['va_metrics'][0]]
4040
return primary_metric, metrics
4141

@@ -208,7 +208,7 @@ def build_record(self, fields, **args):
208208

209209

210210

211-
class OpenNIRPyterrierReRanker(pyterrier.transformer.EstimatorBase):
211+
class OpenNIRPyterrierReRanker(pyterrier.Estimator):
212212
"""
213213
Provides an interface for OpenNIR-style ranking models to act
214214
as re-rankers in pyterrier pipelines.

0 commit comments

Comments
 (0)