Skip to content

Commit

Permalink
rem mlutils dep
Browse files Browse the repository at this point in the history
  • Loading branch information
Vaibhavdixit02 committed Oct 4, 2024
1 parent 3ad81f7 commit 51eaf46
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 9 deletions.
6 changes: 0 additions & 6 deletions Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ LBFGSB = "5be7bae1-8223-5378-bac3-9e7378a2f6e6"
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
LoggingExtras = "e6f89c97-d47a-5376-807f-9c37f3926c36"
MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
OptimizationBase = "bca83a33-5cc9-4baa-983d-23429ab6bcbb"
Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
ProgressLogging = "33c8b6b6-d38a-422a-b730-caa89a2f386c"
Expand All @@ -29,16 +28,11 @@ LBFGSB = "0.4.1"
LinearAlgebra = "1.10"
Logging = "1.10"
LoggingExtras = "0.4, 1"
MLUtils = "0.4.4"
OptimizationBase = "2"
Printf = "1.10"
ProgressLogging = "0.1"
Reexport = "1.2"
SciMLBase = "2.39.0"
SparseArrays = "1.10"
Symbolics = "5.12"
TerminalLoggers = "0.1"
julia = "1.9"

[extras]
Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"
4 changes: 1 addition & 3 deletions src/sophia.jl
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
using Optimization.LinearAlgebra, MLUtils

struct Sophia
η::Float64
βs::Tuple{Float64, Float64}
Expand Down Expand Up @@ -64,7 +62,7 @@ function SciMLBase.__solve(cache::OptimizationCache{

maxiters = Optimization._check_and_convert_maxiters(cache.solver_args.maxiters)

if cache.p isa MLUtils.DataLoader
if OptimizationBase.isa_dataiterator(cache.p)
data = cache.p
dataiterate = true
else
Expand Down

0 comments on commit 51eaf46

Please sign in to comment.