From 061b7f647aa88cee4135fbe91ea95f5f19be4003 Mon Sep 17 00:00:00 2001 From: Abe Levitan Date: Tue, 3 Aug 2021 19:34:08 -0400 Subject: [PATCH] '.' --- CDTools/tools/polarization/polarization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CDTools/tools/polarization/polarization.py b/CDTools/tools/polarization/polarization.py index af5b0b5..7294c80 100644 --- a/CDTools/tools/polarization/polarization.py +++ b/CDTools/tools/polarization/polarization.py @@ -29,7 +29,7 @@ def apply_linear_polarizer(probe, polarizer, multiple_modes=True, transpose=True (N)(P)x2x1xMxL """ - if len(polarizer) == 1: + if len(polarizer.shape) == 1: theta = math.radians(polarizer) jones_matrices = t.tensor([[(cos(theta)) ** 2, sin(2 * theta) / 2], [sin(2 * theta) / 2, sin(theta) ** 2]]).to(dtype=t.cfloat) else: