Include default when complaining about invalid solver types
This commit is contained in:
parent
779346882f
commit
66bbdefcc0
|
|
@ -1555,7 +1555,7 @@ def sample_er_sde(
|
|||
s_in = x.new_ones(x.shape[:1])
|
||||
|
||||
if solver_type not in {"default", "sde", "ersde"}:
|
||||
raise ValueError("Bad solver_type, must be one of ersde or sde")
|
||||
raise ValueError("Bad solver_type, must be one of default, ersde or sde")
|
||||
if noise_scaler is None:
|
||||
if solver_type == "sde":
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue