Merge pull request #52 from SciML/diffeqbase

use DiffEqBase problem definition

Author: ChrisRackauckas

Project.toml

@@ -4,6 +4,7 @@ authors = ["Vaibhavdixit02 <[email protected]>"]
 version = "0.2.2"
 
 [deps]
+ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
 ConsoleProgressMonitor = "88cd18e8-d9cc-4ea6-8889-5259c0d15c8b"
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
 DiffResults = "163ba53b-c6d8-5494-b064-1a9d43ac40c5"
@@ -15,6 +16,7 @@ LoggingExtras = "e6f89c97-d47a-5376-807f-9c37f3926c36"
 Optim = "429524aa-4258-5aef-a3af-852621145aeb"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 ProgressLogging = "33c8b6b6-d38a-422a-b730-caa89a2f386c"
+Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
 TerminalLoggers = "5d786b92-1e48-4d6f-9151-6b4477ca9bed"

README.md

@@ -20,20 +20,20 @@ unified interface.
  x0 = zeros(2)
  p  = [1.0,100.0]
 
- prob = OptimizationProblem(rosenbrock,x0,p=p)
+ prob = OptimizationProblem(rosenbrock,x0,p)
  sol = solve(prob,NelderMead())
 
 
  using BlackBoxOptim
- prob = OptimizationProblem(rosenbrock, x0, p = p, lb = [-1.0,-1.0], ub = [1.0,1.0])
+ prob = OptimizationProblem(rosenbrock, x0, p, lb = [-1.0,-1.0], ub = [1.0,1.0])
  sol = solve(prob,BBO())
 ```
 
 ```julia
- f = OptimizationFunction(rosenbrock, x0, GalacticOptim.AutoForwardDiff(), p=p) 
- prob = OptimizationProblem(f, x0, p = p)
+ f = OptimizationFunction(rosenbrock, x0, GalacticOptim.AutoForwardDiff(), p)
+ prob = OptimizationProblem(f, x0, p)
  sol = solve(prob,BFGS())
-``` 
+```
 
 ```julia
  prob = OptimizationProblem(f, x0, p = p, lb = [-1.0,-1.0], ub = [1.0,1.0])
@@ -58,17 +58,15 @@ choice.
 ### API Documentation
 
 ```julia
-OptimizationFunction(f, x, AutoForwardDiff();
-                     p = DiffEqBase.NullParameters(),
+OptimizationFunction(f, x, AutoForwardDiff(), p = DiffEqBase.NullParameters();
                      grad = nothing,
                      hes = nothing,
                      hv = nothing,
                      chunksize = 1)
 ```
 
 ```julia
-OptimizationProblem(f, x;
-                    p = DiffEqBase.NullParameters(),
+OptimizationProblem(f, x, p = DiffEqBase.NullParameters(),;
                     lb = nothing,
                     ub = nothing)
 ```

src/GalacticOptim.jl

@@ -1,15 +1,19 @@
 module GalacticOptim
 
-using DiffEqBase, Requires  
+using Reexport
+@reexport using DiffEqBase
+using Requires
 using DiffResults, ForwardDiff, Zygote, ReverseDiff, Tracker, FiniteDiff
 using Optim, Flux
 using Logging, ProgressLogging, Printf, ConsoleProgressMonitor, TerminalLoggers, LoggingExtras
+using ArrayInterface
+
+using ForwardDiff: DEFAULT_CHUNK_THRESHOLD
+import DiffEqBase: OptimizationProblem, OptimizationFunction, AbstractADType
 
-include("problem.jl")
 include("solve.jl")
 include("function.jl")
 
-export OptimizationProblem, OptimizationFunction
 export solve
 
 export BBO, CMAEvolutionStrategyOpt