From 332541c9515ca9376c98b45449fa99771408d5f2 Mon Sep 17 00:00:00 2001 From: Abe Levitan Date: Tue, 3 Aug 2021 03:14:00 -0400 Subject: [PATCH] '.' --- CDTools/datasets/polarized_ptycho_2d_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CDTools/datasets/polarized_ptycho_2d_dataset.py b/CDTools/datasets/polarized_ptycho_2d_dataset.py index 93deb31..62c3a25 100644 --- a/CDTools/datasets/polarized_ptycho_2d_dataset.py +++ b/CDTools/datasets/polarized_ptycho_2d_dataset.py @@ -68,7 +68,7 @@ class PolarizedPtycho2DDataset(Ptycho2DDataset): polarizer = [] analyzer = [] - for k in t.tensor(translations).shape[0]: + for k in range(t.tensor(translations).shape[0]): polarizer.append((k//3)%3) analyzer.append((k%3)) self.polarizer = t.tensor(polarizer)