z_position_list should accept a list

mehta-lab · Jul 11, 2023 · 02565bf · 02565bf
1 parent 9b4c27e
commit 02565bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/models/test_isotropic_thin_3d.py b/tests/models/test_isotropic_thin_3d.py
@@ -8,7 +8,7 @@ def test_calculate_transfer_function(axial_flip):
     Hu, Hp = isotropic_thin_3d.calculate_transfer_function(
         yx_shape=(100, 101),
         yx_pixel_size=6.5 / 40,
-        z_position_list=torch.tensor([-1, 0, 1]),
+        z_position_list=[-1, 0, 1],
         wavelength_illumination=0.5,
         index_of_refraction_media=1.0,
         numerical_aperture_illumination=0.4,

diff --git a/waveorder/models/isotropic_thin_3d.py b/waveorder/models/isotropic_thin_3d.py
@@ -41,7 +41,7 @@ def calculate_transfer_function(
     axial_flip=False,
 ):
     if axial_flip:
-        z_position_list = torch.flip(z_position_list, dims=(0,))
+        z_position_list = torch.flip(torch.tensor(z_position_list), dims=(0,))
 
     radial_frequencies = util.generate_radial_frequencies(
         yx_shape, yx_pixel_size