From f8bebd0436aeb468a2168fc6af9a6e079f52fa59 Mon Sep 17 00:00:00 2001 From: gnzng Date: Tue, 4 Mar 2025 14:21:51 -0800 Subject: [PATCH] typo, factor -> factors --- src/cdtools/tools/initializers/initializers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cdtools/tools/initializers/initializers.py b/src/cdtools/tools/initializers/initializers.py index 7e727af..bdb91b9 100644 --- a/src/cdtools/tools/initializers/initializers.py +++ b/src/cdtools/tools/initializers/initializers.py @@ -234,7 +234,7 @@ def gaussian_probe(dataset, basis, shape, sigma, propagation_distance=0, polariz polarizer = dataset.polarizer.tolist() analyzer = dataset.analyzer.tolist() factors = [(math.cos(math.radians(polarizer[idx] - analyzer[idx])))**2 for idx in range(len(dataset)) if (abs(polarizer[idx] - analyzer[idx]) > 5)] - avg_intensities = [t.sum(dataset[idx][1]) / factor[idx] for idx in range(len(dataset))] + avg_intensities = [t.sum(dataset[idx][1]) / factors[idx] for idx in range(len(dataset))] avg_intensity = t.mean(t.tensor(avg_intensities)) probe_intensity = t.sum(t.abs(probe)**2)