fix: wrap doc strings for bessel, bounds, and box

jax_galsim/bessel.py

@@ -105,9 +105,9 @@ def si(x):
     )
 
 
+@implements(_galsim.bessel.kv)
 @jax.jit
 def kv(nu, x):
-    """Modified Bessel 2nd kind"""
     nu = 1.0 * nu
     x = 1.0 * x
     return _tfp_bessel_kve(nu, x) / jnp.exp(jnp.abs(x))

jax_galsim/bounds.py

@@ -96,12 +96,8 @@ def _parse_args(self, *args, **kwargs):
             self._isdefined = False
 
     @property
+    @implements(_galsim.Bounds.true_center)
     def true_center(self):
-        """The central position of the `Bounds` as a `PositionD`.
-
-        This is always (xmax + xmin)/2., (ymax + ymin)/2., even for integer `BoundsI`, where
-        this may not necessarily be an integer `PositionI`.
-        """
         if not self.isDefined():
             raise _galsim.GalSimUndefinedBoundsError(
                 "true_center is invalid for an undefined Bounds"

jax_galsim/box.py

@@ -28,13 +28,13 @@ def _maxL(self):
         return jnp.maximum(self.width, self.height)
 
     @property
+    @implements(_galsim.Box.width)
     def width(self):
-        """The width of the `Box`."""
         return self.params["width"]
 
     @property
+    @implements(_galsim.Box.height)
     def height(self):
-        """The height of the `Box`."""
         return self.params["height"]
 
     def __hash__(self):
@@ -115,7 +115,6 @@ def tree_unflatten(cls, aux_data, children):
             **aux_data,
         )
 
-    @implements(_galsim.Box._shoot)
     def _shoot(self, photons, rng):
         ud = UniformDeviate(rng)
 
@@ -134,6 +133,7 @@ def __init__(self, scale, flux=1.0, gsparams=None):
         )
 
     @property
+    @implements(_galsim.Pixel.scale)
     def scale(self):
         """The linear scale size of the `Pixel`."""
         return self.width