Properly handle Bayesian model import failure (#207)

* Properly handle Bayesian model import failure * Fix incompat tensorflow version --------- Co-authored-by: Pierre-Paul De Breuck <[email protected]>
ppdebreuck · Apr 2, 2024 · 1d08ebf · 1d08ebf
1 parent 024d720
commit 1d08ebf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modnet/models/bayesian.py b/modnet/models/bayesian.py
@@ -15,7 +15,7 @@
 try:
     import tensorflow_probability as tfp
 except ImportError:
-    raise RuntimeError(
+    raise ImportError(
         "`tensorflow-probability` is required for Bayesian models: install modnet[bayesian]."
     )