Skip to content

Commit 703b1f4

Browse files
Jammy2211Jammy2211
authored andcommitted
fix PSF convlution in numpy
1 parent eeeb4ff commit 703b1f4

3 files changed

Lines changed: 229 additions & 140 deletions

File tree

autoarray/dataset/imaging/simulator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def __init__(
100100
self.noise_seed = noise_seed
101101

102102
def via_image_from(
103-
self, image: Array2D, over_sample_size: Optional[Union[int, np.ndarray]] = None
103+
self, image: Array2D, over_sample_size: Optional[Union[int, np.ndarray]] = None, xp=np
104104
) -> Imaging:
105105
"""
106106
Simulate an `Imaging` dataset from an input image.
@@ -126,7 +126,7 @@ def via_image_from(
126126
pixel_scales=image.pixel_scales,
127127
)
128128

129-
image = self.psf.convolved_image_from(image=image, blurring_image=None)
129+
image = self.psf.convolved_image_from(image=image, blurring_image=None, xp=xp)
130130

131131
image = image + background_sky_map
132132

0 commit comments

Comments
 (0)