We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bb5a866 commit 46aaf8cCopy full SHA for 46aaf8c
python/nutpie/compile_pymc.py
@@ -499,6 +499,11 @@ def compile_pymc_model(
499
"and restart your kernel in case you are in an interactive session."
500
)
501
502
+ if gradient_backend is not None:
503
+ gradient_backend = gradient_backend.lower() # type: ignore[assignment]
504
+ if backend is not None:
505
+ backend = backend.lower() # type: ignore[assignment]
506
+
507
from pymc.model.transform.optimization import freeze_dims_and_data
508
from pymc.initial_point import make_initial_point_fn
509
0 commit comments