From 40afea84b947b2f5652d879ccd922dd8fa45aff4 Mon Sep 17 00:00:00 2001
From: Andrew Parry <a.parry.1@research.gla.ac.uk>
Date: Sun, 15 Dec 2024 15:48:16 +0000
Subject: [PATCH] fixing import errors

---
 rankers/pyterrier/dot/dot.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/rankers/pyterrier/dot/dot.py b/rankers/pyterrier/dot/dot.py
index 2af6558..52d24a8 100644
--- 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