🔧 Properly assign values to dataframe. (#272)

openvinotoolkit · Apr 25, 2022 · d0fb1f8 · d0fb1f8
1 parent 24ccb58
commit d0fb1f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/anomalib/data/folder.py b/anomalib/data/folder.py
@@ -146,7 +146,7 @@ def make_dataset(
         samples["mask_path"] = ""
         for index, row in samples.iterrows():
             if row.label_index == 1:
-                samples["mask_path"][index] = str(mask_dir / row.image_path.name)
+                samples.loc[index, "mask_path"] = str(mask_dir / row.image_path.name)
 
     # Ensure the pathlib objects are converted to str.
     # This is because torch dataloader doesn't like pathlib.