added kwargs for convolve

LouisDesdoigts · Oct 14, 2024 · cb89a83 · cb89a83
1 parent 5089344
commit cb89a83
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/dLux/psfs.py b/src/dLux/psfs.py
@@ -79,7 +79,7 @@ def downsample(self: PSF, n: int) -> PSF:
         """
         return self.set("data", dlu.downsample(self.data, n, mean=False))
 
-    def convolve(self: PSF, other: Array) -> PSF:
+    def convolve(self: PSF, other: Array, **kwargs) -> PSF:
         """
         Convolves the psf with some input array.
 
@@ -93,7 +93,9 @@ def convolve(self: PSF, other: Array) -> PSF:
         psf : PSF
             The convolved psf.
         """
-        return self.set("data", convolve(self.data, other, mode="same"))
+        return self.set(
+            "data", convolve(self.data, other, mode="same", **kwargs)
+        )
 
     def rotate(self: PSF, angle: float, order: int = 1) -> PSF:
         """