From b2c298fcf0daffdbd26267b7f462b7992748f9b3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Benjamin=20Clavi=C3=A9?= Date: Mon, 18 Mar 2024 19:03:14 +0100 Subject: [PATCH] lint --- ragatouille/models/torch_kmeans.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/ragatouille/models/torch_kmeans.py b/ragatouille/models/torch_kmeans.py index 4e3135c..30db7bc 100644 --- a/ragatouille/models/torch_kmeans.py +++ b/ragatouille/models/torch_kmeans.py @@ -1,9 +1,6 @@ -import time import torch from fast_pytorch_kmeans import KMeans -import math -import tqdm def _train_kmeans(self, sample, shared_lists): # noqa: ARG001