From 02565bfad7859f3facafb1e66b700db99f2fca80 Mon Sep 17 00:00:00 2001 From: Talon Chandler Date: Tue, 11 Jul 2023 14:39:50 -0700 Subject: [PATCH] `z_position_list` should accept a list --- tests/models/test_isotropic_thin_3d.py | 2 +- waveorder/models/isotropic_thin_3d.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/test_isotropic_thin_3d.py b/tests/models/test_isotropic_thin_3d.py index 7af34be..aa4acf9 100644 --- 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 index 4cfa954..93260cd 100644 --- 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