Update src/data/dataset_csv.py

src/data/dataset_csv.py

@@ -12,7 +12,7 @@ import nibabel as nib
 import json
 
 from tqdm import tqdm
-from .MRI_load import load_mris
+#from .MRI_load import load_mris
 from collections import defaultdict
 from time import time
 from copy import deepcopy
@@ -142,7 +142,7 @@ class CSVDataset:
                 elif img_mode == 2: 
                     # load MRIs and use swinunetr model to get the embeddings
                     print('img_mode is 2')
-                    embedding_dict = load_mris.get_emb('filename', df, arch=arch, transforms=transforms, stripped=stripped)
+                    #embedding_dict = load_mris.get_emb('filename', df, arch=arch, transforms=transforms, stripped=stripped)
                     mri_embeddings = []
                     for index, row in df.iterrows():
                         filename = row['filename']