Skip to content

Commit 18df01c

Browse files
Uncomment check for torch backend.
1 parent 1958982 commit 18df01c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayesflow/wrappers/mamba/mamba_block.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,8 @@ def __init__(
5555

5656
super().__init__(**keras_kwargs(kwargs))
5757

58-
# if keras.backend.backend() != "torch":
59-
# raise RuntimeError("Mamba is only available using torch backend.")
58+
if keras.backend.backend() != "torch":
59+
raise RuntimeError("Mamba is only available using torch backend.")
6060

6161
try:
6262
from mamba_ssm import Mamba

0 commit comments

Comments
 (0)