add utils fix

in process dataset, add code handling added 'Test' column

utils/utils.py

@@ -520,7 +520,11 @@ def process_dataset_in_mem(embeddings_df: pd.DataFrame,
         balanced_train_images = balanced_train_df['Images'].tolist()
         
         # Convert test set back to numpy array and list
-        test_embeddings = test_df.drop(columns=['Label', 'Images']).to_numpy()
+        if 'Test' in test_df:
+            test_embeddings = test_df.drop(columns=['Label', 'Images', 'Test']).to_numpy()
+        else:
+            test_embeddings = test_df.drop(columns=['Label', 'Images']).to_numpy()
+
         test_labels = test_df['Label'].tolist()
         test_images = test_df['Images'].tolist()