Fixing a bug related to odak.learn.wave.gerchberg_saxton().

kaanaksit · Apr 23, 2022 · 9df3b6a · 9df3b6a
1 parent c8af57e
commit 9df3b6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/odak/learn/wave/classical.py b/odak/learn/wave/classical.py
@@ -287,7 +287,7 @@ def gerchberg_saxton(field, n_iterations, distance, dx, wavelength, slm_range=6.
         hologram, _ = produce_phase_only_slm_pattern(hologram, slm_range)
         reconstruction = propagate_beam(
             hologram, k, distance, dx, wavelength, propagation_type)
-        reconstruction = set_amplitude(hologram, field)
+        reconstruction = set_amplitude(reconstruction, field)
     reconstruction = propagate_beam(
         hologram, k, distance, dx, wavelength, propagation_type)
     return hologram, reconstruction