From 9cce3612cd8fa70b7f8e2cea86dcf34120057ee4 Mon Sep 17 00:00:00 2001 From: bclavie Date: Fri, 23 Feb 2024 12:06:38 +0100 Subject: [PATCH] isort --- ragatouille/models/colbert.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ragatouille/models/colbert.py b/ragatouille/models/colbert.py index 303a24c..534a952 100644 --- a/ragatouille/models/colbert.py +++ b/ragatouille/models/colbert.py @@ -11,6 +11,7 @@ from colbert import Indexer, IndexUpdater, Searcher, Trainer from colbert.infra import ColBERTConfig, Run, RunConfig from colbert.modeling.checkpoint import Checkpoint + from ragatouille.models.base import LateInteractionModel # TODO: Move all bsize related calcs to `_set_bsize()`