Resize fix for new torch. (#13)

tests/test_simulate.py

@@ -0,0 +1,26 @@
+from waveprop.simulation import FarFieldSimulator
+import torch
+from waveprop.devices import sensor_dict, SensorParam
+
+
+def test_far_field_simulator():
+
+    sensor = "rpi_hq"
+    sensor_shape = sensor_dict[sensor][SensorParam.SHAPE]
+
+
+    sim = FarFieldSimulator(
+        object_height=30e-2,
+        scene2mask=30e-2,
+        mask2sensor=4e-3,
+        sensor=sensor,
+        output_dim=sensor_shape,
+    )
+
+    obj = torch.rand(1, 1, 256, 256)
+    image = sim.propagate(obj)
+    assert image.shape == (1, 1, *sensor_shape)
+
+
+if __name__ == "__main__":
+    test_far_field_simulator()

waveprop/util.py

@@ -560,6 +560,7 @@ def prepare_object_plane(
     object_height_pix = int(np.round(object_height / scene_dim[1] * sensor_dim[1]))
     scaling = object_height_pix / input_dim[1]
     object_dim = tuple((np.round(input_dim * scaling)).astype(int))
+    object_dim = (int(object_dim[0]), int(object_dim[1]))
 
     if torch.is_tensor(obj):
         object_plane = resize_torch(obj, size=object_dim, antialias=True)