From 24475dbdac5b1e0f687d6c588cee33b9d37d2b07 Mon Sep 17 00:00:00 2001 From: Abe Levitan Date: Tue, 3 Aug 2021 20:14:15 -0400 Subject: [PATCH] '.' --- CDTools/datasets/polarized_ptycho_2d_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/CDTools/datasets/polarized_ptycho_2d_dataset.py b/CDTools/datasets/polarized_ptycho_2d_dataset.py index 62c3a25..b1e22a7 100644 --- a/CDTools/datasets/polarized_ptycho_2d_dataset.py +++ b/CDTools/datasets/polarized_ptycho_2d_dataset.py @@ -69,8 +69,8 @@ class PolarizedPtycho2DDataset(Ptycho2DDataset): polarizer = [] analyzer = [] for k in range(t.tensor(translations).shape[0]): - polarizer.append((k//3)%3) - analyzer.append((k%3)) + polarizer.append((k//3)%3 * 45) + analyzer.append((k%3 * 45)) self.polarizer = t.tensor(polarizer) self.analyzer = t.tensor(analyzer)