diff --git a/mu_map/dataset/mock.py b/mu_map/dataset/mock.py index 9f33b3c900f0a85b053a2e7efabedb04de4e783b..8cfd95122c21b805f574abc44a1155fd7a49ffcc 100644 --- a/mu_map/dataset/mock.py +++ b/mu_map/dataset/mock.py @@ -11,7 +11,7 @@ class MuMapMockDataset(MuMapDataset): recon, mu_map = super().__getitem__(0) recon = recon[:, :32, :, :] mu_map = mu_map[:, :32, :, :] - return recon, mu_map + return recon, mu_map / 40206.0 def __len__(self): return self.len