Removing deprecation warning and jitting predict (#120)

* removing deprecation warning and jitting predict

* adding news

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

news/120.feature

@@ -0,0 +1,2 @@
+Removed deprecation warning from ``predict`` method and wrapped it in a
+``jax.jit`` in order to support interactive use.

src/tinygp/gp.py

@@ -213,6 +213,11 @@ def condition(
 
         return ConditionResult(log_prob, gp)
 
+    @partial(
+        jax.jit,
+        static_argnums=(0,),
+        static_argnames=("include_mean", "return_var", "return_cov"),
+    )
     def predict(
         self,
         y: JAXArray,
@@ -250,14 +255,6 @@ def predict(
             returned with shape ``(N_test,)`` or ``(N_test, N_test)``
             respectively.
         """
-        import warnings
-
-        warnings.warn(
-            "The 'predict' method is deprecated and 'condition' should be preferred",
-            DeprecationWarning,
-            stacklevel=2,
-        )
-
         _, cond = self.condition(
             y, X_test, kernel=kernel, include_mean=include_mean
         )