Fix scipy

JaxGaussianProcesses · Nov 30, 2023 · 9e35e3e · 9e35e3e
1 parent 3e475d8
commit 9e35e3e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gpjax/__init__.py b/gpjax/__init__.py
@@ -29,7 +29,10 @@
 )
 from gpjax.citation import cite
 from gpjax.dataset import Dataset
-from gpjax.fit import fit
+from gpjax.fit import (
+    fit,
+    fit_scipy,
+)
 
 __license__ = "MIT"
 __description__ = "Didactic Gaussian processes in JAX"
@@ -52,4 +55,5 @@
     "fit",
     "Module",
     "param_field",
+    "fit_scipy",
 ]