Skip to content

Commit fc86d4d

Browse files
committed
SIR example: use tensorflow as default backend since jax currently fails
1 parent 4a23ded commit fc86d4d

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

examples/SIR_PosteriorEstimation.ipynb

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,12 @@
2323
"import numpy as np\n",
2424
"import pandas as pd\n",
2525
"\n",
26+
"# ensure the backend is set\n",
27+
"import os\n",
28+
"if \"KERAS_BACKEND\" not in os.environ:\n",
29+
" # set this to \"torch\", \"tensorflow\", or \"jax\"\n",
30+
" os.environ[\"KERAS_BACKEND\"] = \"tensorflow\"\n",
31+
"\n",
2632
"import keras\n",
2733
"\n",
2834
"# For BayesFlow devs: this ensures that the latest dev version can be found\n",

0 commit comments

Comments
 (0)