lint fix

krypticmouse · Jul 5, 2024 · 23ef62b · 23ef62b
1 parent cabfbf9
commit 23ef62b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pirate/retrievers/bi_encoder.py b/pirate/retrievers/bi_encoder.py
@@ -1,5 +1,6 @@
 import torch
 
+from tqdm import tqdm
 from loguru import logger
 from typing import Optional
 from sentence_transformers import SentenceTransformer