Skip to content

Commit 46aaf8c

Browse files
committed
fix: allow upper case backend string
fixes #167
1 parent bb5a866 commit 46aaf8c

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

python/nutpie/compile_pymc.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -499,6 +499,11 @@ def compile_pymc_model(
499499
"and restart your kernel in case you are in an interactive session."
500500
)
501501

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+
502507
from pymc.model.transform.optimization import freeze_dims_and_data
503508
from pymc.initial_point import make_initial_point_fn
504509

0 commit comments

Comments
 (0)