diff --git a/function.py b/function.py
index bea93c8..3c7a9ad 100644
--- a/function.py
+++ b/function.py
@@ -1,6 +1,6 @@
 # function.py
 import numpy as np
-from collections import defaultdict
+
 
 """
 挑战说明:
@@ -78,7 +78,7 @@ def classify0(inX, dataSet, labels, k=3):
     distances = np.array([mahalanobis_distance(inX, x, cov_inv, mean) for x in dataSet])
     
     # 获取距离最近的 k 个样本的索引
-    sortedDistIndices = argpartition(distances, k)[:k]
+    sortedDistIndices = np.argpartition(distances, k)[:k]
     
     # 获取这些样本对应的标签
     classCount = {}