Bump pyterrier api (#14)

* Update __init__.py * require pyterrier 0.9 (Nov' 22)
terrierteam · Oct 3, 2024 · 8caa9a2 · 8caa9a2
1 parent bc506bf
commit 8caa9a2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/pyterrier_t5/__init__.py b/pyterrier_t5/__init__.py
@@ -9,12 +9,11 @@
 import torch
 from torch.nn import functional as F
 from transformers import T5Config, T5Tokenizer, T5ForConditionalGeneration, MT5ForConditionalGeneration
-from pyterrier.transformer import TransformerBase
 from typing import List
 import re
 
 
-class MonoT5ReRanker(TransformerBase):
+class MonoT5ReRanker(pt.Transformer):
     def __init__(self, 
                  tok_model='t5-base',
                  model='castorini/monot5-base-msmarco',
@@ -66,7 +65,7 @@ def transform(self, run):
         return run
 
 
-class DuoT5ReRanker(TransformerBase):
+class DuoT5ReRanker(pt.Transformer):
     def __init__(self, 
                  tok_model='t5-base',
                  model='castorini/duot5-base-msmarco',

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,4 @@
-python-terrier>=0.5.0
-pandas
+python-terrier>=0.9.0
 transformers>=4.0.0
 torch
 sentencepiece>=0.1.95