Skip to content

Commit f7fe860

Browse files
committed
fix error type in diffusion model [no ci]
1 parent 3c9797d commit f7fe860

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayesflow/experimental/diffusion_model/diffusion_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,10 +91,10 @@ def __init__(
9191
super().__init__(base_distribution="normal", **kwargs)
9292

9393
if prediction_type not in ["noise", "velocity", "F", "x"]:
94-
raise TypeError(f"Unknown prediction type: {prediction_type}")
94+
raise ValueError(f"Unknown prediction type: {prediction_type}")
9595

9696
if loss_type not in ["noise", "velocity", "F"]:
97-
raise TypeError(f"Unknown loss type: {loss_type}")
97+
raise ValueError(f"Unknown loss type: {loss_type}")
9898

9999
if loss_type != "noise":
100100
logging.warning(

0 commit comments

Comments
 (0)