remove jax static argnums in bilateral blur

ASEM000 · Apr 10, 2024 · 80f2cc2 · 80f2cc2
1 parent bc2509c
commit 80f2cc2
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/serket/_src/image/filter.py b/serket/_src/image/filter.py
@@ -615,7 +615,6 @@ def fft_elastic_transform_2d(
     return map_coordinates(image, (ny, nx), order=1, mode="nearest").reshape(r, c)
 
 
-@ft.partial(jax.jit, inline=True, static_argnums=1)
 def bilateral_blur_2d(
     image: HWArray,
     kernel_size: tuple[int, int],
@@ -663,7 +662,6 @@ def bilateral_blur_2d(array):
     return bilateral_blur_2d(image)
 
 
-@ft.partial(jax.jit, inline=True, static_argnums=2)
 def joint_bilateral_blur_2d(
     image: HWArray,
     guidance: HWArray,