From 3116ca4e0f7d499cb0a26db3bfc2018b2d0530c9 Mon Sep 17 00:00:00 2001
From: Joseph Omar <j.omar@soton.ac.uk>
Date: Mon, 18 Nov 2024 10:18:39 +0000
Subject: [PATCH] fix sklearn/scipy import error

---
 entcl/utils/ncd.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/entcl/utils/ncd.py b/entcl/utils/ncd.py
index 7093397..2fe4a58 100644
--- a/entcl/utils/ncd.py
+++ b/entcl/utils/ncd.py
@@ -5,7 +5,7 @@ from loguru import logger
 import numpy as np
 from sklearn.cluster import KMeans
 from sklearn.metrics import confusion_matrix
-from sklearn.utils.linear_assignment_ import linear_sum_assignment
+from scipy.optimize import linear_sum_assignment
 import torch
 from tqdm import tqdm
 
-- 
GitLab