Fix ptycho_2d_dataset

2026-09-10 13:32:40 +02:00 · 2022-03-08 10:01:26 -05:00
parent 571fe19ca3
commit ddcc5bbab4
2 changed files with 3 additions and 3 deletions
@@ -165,7 +165,7 @@ class Ptycho2DDataset(CDataset):
            dataset.mask = t.ones(dataset.patterns.shape[-2:]).to(dtype=t.bool)

        try:
-            intensities = cdtdata.get_shot_to_shot_info('intensities')
+            intensities = cdtdata.get_shot_to_shot_info(cxi_file, 'intensities')
            dataset.intensities = t.as_tensor(intensities, dtype=t.float32)
        except KeyError:
            dataset.intensities = None