patch dataset now correctly uses augemntation transforms

mu_map/dataset/patches.py

@@ -65,6 +65,8 @@ class MuMapPatchDataset(MuMapDataset):
         recon = recon[:, z : z + s, y : y + s, x : x + s]
         mu_map = mu_map[:, z : z + s, y : y + s, x : x + s]
 
+        recon, mu_map = self.transform_augmentation(recon, mu_map)
+
         return recon, mu_map
 
     def __len__(self):