diff --git a/test/neural_de.jl b/test/neural_de.jl index 13ed03541e..6a3668fa52 100644 --- a/test/neural_de.jl +++ b/test/neural_de.jl @@ -10,4 +10,4 @@ neural_ode(x,dudt,tspan,diffeq_adjoint,Tsit5(),saveat=0.1) neural_ode(x,dudt,tspan,diffeq_fd,Tsit5(),saveat=0.1) neural_ode(x,dudt,tspan,diffeq_rd,Tsit5(),saveat=0.1) -neural_msde(x,dudt,[0.1,0.1],tspan,SOSRI(),save_everystep=false,save_start=false) +neural_msde(x,dudt,[0.1,0.1],tspan,SOSRI(),saveat=0.1)