From 51eaf4640519814c6fc1afb04a74d570ea842d23 Mon Sep 17 00:00:00 2001
From: Vaibhav Dixit <vaibhavyashdixit@gmail.com>
Date: Fri, 4 Oct 2024 13:30:36 -0400
Subject: [PATCH] rem mlutils dep

---
 Project.toml  | 6 ------
 src/sophia.jl | 4 +---
 2 files changed, 1 insertion(+), 9 deletions(-)

diff --git a/Project.toml b/Project.toml
index 105e65e21..e484ab75c 100644
--- a/Project.toml
+++ b/Project.toml
@@ -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"
@@ -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"
diff --git a/src/sophia.jl b/src/sophia.jl
index 5419b87d7..88b0812c3 100644
--- a/src/sophia.jl
+++ b/src/sophia.jl
@@ -1,5 +1,3 @@
-using Optimization.LinearAlgebra, MLUtils
-
 struct Sophia
     η::Float64
     βs::Tuple{Float64, Float64}
@@ -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