fix dc term

mehta-lab · Oct 2, 2024 · 902c690 · 902c690
1 parent 2b7fd04
commit 902c690
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/waveorder/optics.py b/waveorder/optics.py
@@ -239,6 +239,9 @@ def rotation_matrix(nu_z, nu_y, nu_x, wavelength):
 
     out = torch.stack((row0, row1, row2), dim=0)
 
+    # KLUDGE: fix the DC term manually, avoiding nan
+    out[..., 0, 0] = torch.tensor([[0, 0], [1, 0], [0, 1]])[..., None]
+
     return torch.nan_to_num(out, nan=0.0)