Fix Bug on -n flag

1 files changed, 3 insertions, 1 deletions

diff --git a/part2.py b/part2.py
index 1d6485c..9a30aad 100755
--- a/part2.py
+++ b/part2.py
@@ -97,8 +97,10 @@ def test_model(gallery_data, probe_data, gallery_label, probe_label, gallery_cam
     nneighbors = np.zeros((ranklist.shape[0],nsize))
     nnshowrank = (np.zeros((ranklist.shape[0],nsize))).astype(object)
     
+    
     for i in range(args.multrank):
-        args.neighbors = test_table[i]
+        if args.multrank!= 1:
+            args.neighbors = test_table[i]
         for probe_idx in range(probe_data.shape[0]):
             row = ranklist[probe_idx]
             n = 0