Fixed default argument for backend

pymc-devs · Oct 11, 2024 · 8b9521c · 8b9521c
1 parent d94472d
commit 8b9521c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/nutpie/compile_pymc.py b/python/nutpie/compile_pymc.py
@@ -354,7 +354,7 @@ def expand(x, **shared):
 def compile_pymc_model(
     model: "pm.Model",
     *,
-    backend: Literal["numba", "jax"] = "numba",
+    backend: Literal["numba", "jax"] | None = None,
     gradient_backend: Literal["pytensor", "jax"] | None = None,
     **kwargs,
 ) -> CompiledModel: