From f44201960ac92be4454ab31a64a853702e3c81b0 Mon Sep 17 00:00:00 2001
From: Hossein Pourbozorg <prbzrg@gmail.com>
Date: Wed, 22 May 2024 15:03:35 +0330
Subject: [PATCH] use polyalg (#395)

* test polyalg

* fix

* rm

* format

* dep on oDE
---
 Project.toml                      | 4 ----
 src/ContinuousNormalizingFlows.jl | 2 --
 src/base_icnf.jl                  | 5 +----
 3 files changed, 1 insertion(+), 10 deletions(-)

diff --git a/Project.toml b/Project.toml
index ae0cbdf4..c693fd27 100644
--- a/Project.toml
+++ b/Project.toml
@@ -11,7 +11,6 @@ ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 ComputationalResources = "ed09eef8-17a6-5b46-8889-db040fac31e3"
 DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
-DifferentialEquations = "0c46a032-eb83-5123-abaf-570d42b7fbaa"
 DifferentiationInterface = "a0c0ee7d-e4b9-4e03-894e-1c5f64a51d63"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DistributionsAD = "ced4e74d-a319-5a8a-b0ac-84af2272839c"
@@ -32,7 +31,6 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SciMLBase = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
 SciMLSensitivity = "1ed8b502-d754-442c-8d5d-10ac956f44a1"
 ScientificTypesBase = "30f210dd-8aff-4c5f-94ba-8e64358c1161"
-Static = "aedffcd0-7271-4cad-89d0-dc628f76c6d3"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
@@ -51,7 +49,6 @@ ComponentArrays = "0.15"
 ComputationalResources = "0.3"
 DataFrames = "1"
 Dates = "1"
-DifferentialEquations = "7"
 DifferentiationInterface = "0.1, 0.2, 0.3, 0.4"
 Distributions = "0.25"
 DistributionsAD = "0.6"
@@ -72,7 +69,6 @@ Random = "1"
 SciMLBase = "2"
 SciMLSensitivity = "7"
 ScientificTypesBase = "3"
-Static = "0.8"
 Statistics = "1"
 Zygote = "0.6"
 julia = "1.9"
diff --git a/src/ContinuousNormalizingFlows.jl b/src/ContinuousNormalizingFlows.jl
index 55b54d46..38fc4607 100644
--- a/src/ContinuousNormalizingFlows.jl
+++ b/src/ContinuousNormalizingFlows.jl
@@ -8,7 +8,6 @@ import AbstractDifferentiation,
     ComputationalResources,
     DataFrames,
     Dates,
-    DifferentialEquations,
     DifferentiationInterface,
     Distributions,
     DistributionsAD,
@@ -27,7 +26,6 @@ import AbstractDifferentiation,
     OrdinaryDiffEq,
     Random,
     ScientificTypesBase,
-    Static,
     SciMLBase,
     SciMLSensitivity,
     Statistics,
diff --git a/src/base_icnf.jl b/src/base_icnf.jl
index 19180ede..16dc5a3d 100644
--- a/src/base_icnf.jl
+++ b/src/base_icnf.jl
@@ -20,10 +20,7 @@ function construct(
     ),
     differentiation_backend::AbstractDifferentiation.AbstractBackend = AbstractDifferentiation.ZygoteBackend(),
     autodiff_backend::ADTypes.AbstractADType = ADTypes.AutoZygote(),
-    sol_kwargs::NamedTuple = (
-        save_everystep = false,
-        alg = OrdinaryDiffEq.Tsit5(; thread = Static.True()),
-    ),
+    sol_kwargs::NamedTuple = (save_everystep = false,),
     rng::Random.AbstractRNG = rng_AT(resource),
     λ₁::AbstractFloat = if aicnf <: Union{RNODE, CondRNODE}
         convert(data_type, 1e-2)