docs: more api + typos

docs/api.rst

@@ -5,4 +5,6 @@ API
 
 .. autofunction:: coverage
 
-.. autofunction:: period_match
+.. autofunction:: period_match
+
+.. autofunction:: phase_coverage

docs/tess.ipynb

@@ -199,7 +199,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "We see here that a portion of the phase space is covered less than two times, meaning that we would not be able to see two transit of this particular exoplanet. Let's compute this portion"
+    "We see here that a portion of the phase space is covered less than two times, meaning that we would not be able to see two transits of this particular exoplanet. Let's compute this portion"
    ]
   },
   {

portrait/__init__.py

@@ -18,4 +18,4 @@ def get_version() -> str:
 
 version: str = get_version()
 
-from portrait.core import coverage, period_match
+from portrait.core import coverage, period_match, phase_coverage

portrait/core.py

@@ -122,11 +122,10 @@ def coverage(time, gap=0.1, precision=1e-3, n=1):
         i.e. with signature :code:`fun(float or array) -> float or array`
     """
 
-    phases = jnp.arange(0, 1, precision)
+    phases = jnp.arange(0.0, 1.0, precision)
     overlap_function = phase_coverage(time, phases, gap=gap)
 
-    jax.jit
-
+    @jax.jit
     def fun(period):
         """Compute the phase coverage for a given period