fixing import errors

Parry-Parry · Dec 15, 2024 · 40afea8 · 40afea8
1 parent 9f0fc70
commit 40afea8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/rankers/pyterrier/dot/dot.py b/rankers/pyterrier/dot/dot.py
@@ -5,8 +5,7 @@
     AutoModel,
     AutoTokenizer,
 )
-from ...modelling import Dot, DotConfig
-from ...modelling.dot.dot import Pooler
+from ...modelling.dot.dot import Pooler, Dot, DotConfig
 from typing import Union
 import torch
 import pandas as pd