From b2631d0f6ba8f3b21925a7572262d270bf469c67 Mon Sep 17 00:00:00 2001
From: Jose Esparza <28990958+pebeto@users.noreply.github.com>
Date: Tue, 15 Aug 2023 17:08:23 -0500
Subject: [PATCH] Removing MLJ as a test dependency

---
 Project.toml     | 5 ++---
 test/base.jl     | 2 +-
 test/runtests.jl | 5 ++++-
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/Project.toml b/Project.toml
index 657c698..adb5f64 100644
--- a/Project.toml
+++ b/Project.toml
@@ -10,16 +10,15 @@ MLJModelInterface = "e80e1ace-859a-464e-9ed9-23947d8ae3ea"
 
 [compat]
 MLFlowClient = "0.4.4"
-MLJ = "0.19"
 MLJBase = "0.21.11"
 MLJModelInterface = "1.9.1"
 julia = "1.6"
 
 [extras]
 MLFlowClient = "64a0f543-368b-4a9a-827a-e71edb2a0b83"
-MLJ = "add582a8-e3ab-11e8-2d5e-e98b27df1bc7"
 MLJDecisionTreeInterface = "c6f25543-311c-4c74-83dc-3ea6d1015661"
+MLJModels = "d491faf4-2d78-11e9-2867-c94bc002c0b7"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test", "MLFlowClient", "MLJ", "MLJDecisionTreeInterface"]
+test = ["Test", "MLFlowClient", "MLJModels", "MLJDecisionTreeInterface"]
diff --git a/test/base.jl b/test/base.jl
index 944852d..90474d1 100644
--- a/test/base.jl
+++ b/test/base.jl
@@ -18,7 +18,7 @@
     end
 
     @testset "save" begin
-        run = MLJ.save(logger, dtc_machine)
+        run = MLJBase.save(logger, dtc_machine)
         @test typeof(run) == MLFlowRun
         @test listartifacts(logger.client, run) |> length == 1
     end
diff --git a/test/runtests.jl b/test/runtests.jl
index 6a5d159..20f209c 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -1,8 +1,11 @@
 using Test
 
-using MLJ
 using MLJFlow
+
+using MLJBase
+using MLJModels
 using MLFlowClient
+using MLJModelInterface
 
 include("base.jl")
 include("types.jl")