diff --git a/function.py b/function.py
index 6869245..0ab1bd4 100644
--- a/function.py
+++ b/function.py
@@ -134,9 +134,9 @@ def img_to_vector(filename):
 
 if __name__ == "__main__":
     # 测试代码
-   train_data, train_labels = load_dataset("trainingDigits")
+    train_data, train_labels = load_dataset("trainingDigits")
 
-   test_data = (
+    test_data = (
         "00000000000000110000000000000000"
         "00000000000011111000000000000000"
         "00000000000011111100000000000000"
@@ -179,4 +179,4 @@ if __name__ == "__main__":
     
     # 使用 KDTree 加速的 KNN 进行预测
     predicted_label_kdtree = classify(test_vector, train_data, train_labels, k=5, use_kdtree=True)
-    print(f"KDTree 加速后的预测标签: {predicted_label_kdtree}")
\ No newline at end of file
+    print(f"KDTree 加速后的预测标签: {predicted_label_kdtree}")