fix tests

oscardssmith · oscardssmith · commit c406d040328d · 2025-04-29T01:34:43.000-04:00
diff --git a/test/diffeqfluxtests.jl b/test/diffeqfluxtests.jl
@@ -1,4 +1,4 @@
-using OrdinaryDiffEq, DiffEqFlux, Lux, Optimization, OptimizationOptimJL,
+using OrdinaryDiffEqTsit5, DiffEqFlux, Lux, Optimization, OptimizationOptimJL,
       OptimizationOptimisers, ForwardDiff, ComponentArrays, Random
 rng = Random.default_rng()
 Random.seed!(123)
diff --git a/test/downstream/Project.toml b/test/downstream/Project.toml
@@ -1,4 +1,3 @@
 [deps]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 DiffEqFlux = "aae7a2af-3d4f-5e19-a356-7da93b79d9d0"
-OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
diff --git a/test/downstream/gpu_neural_ode.jl b/test/downstream/gpu_neural_ode.jl
@@ -1,4 +1,4 @@
-using DiffEqFlux, OrdinaryDiffEq, Flux, CUDA
+using DiffEqFlux, OrdinaryDiffEqTsit5, Flux, CUDA
 CUDA.allowscalar(false) # Makes sure no slow operations are occurring
 
 # Generate Data

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-using OrdinaryDiffEq, DiffEqFlux, Lux, Optimization, OptimizationOptimJL,`
	`1`	`+using OrdinaryDiffEqTsit5, DiffEqFlux, Lux, Optimization, OptimizationOptimJL,`
`2`	`2`	`OptimizationOptimisers, ForwardDiff, ComponentArrays, Random`
`3`	`3`	`rng = Random.default_rng()`
`4`	`4`	`Random.seed!(123)`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-using DiffEqFlux, OrdinaryDiffEq, Flux, CUDA`
	`1`	`+using DiffEqFlux, OrdinaryDiffEqTsit5, Flux, CUDA`
`2`	`2`	`CUDA.allowscalar(false) # Makes sure no slow operations are occurring`
`3`	`3`
`4`	`4`	`# Generate Data`