From 3b9a665046a8827effc22583bd77e653e994701c Mon Sep 17 00:00:00 2001
From: pat-alt <altmeyerpat@gmail.com>
Date: Mon, 8 Apr 2024 17:20:01 +0200
Subject: [PATCH] trying to fix error

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

diff --git a/test/Project.toml b/test/Project.toml
index 93ab25a..1665246 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -3,7 +3,8 @@ CondaPkg = "992eb4ea-22a4-4c89-a5bb-47a3300528ab"
 CounterfactualExplanations = "2f13d31b-18db-44c1-bc43-ebaf2cff0be0"
 Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 PackageExtensionCompat = "65ce6f38-6b18-4e1d-a461-8949797d7930"
-PythonCall = "6099a3de-0909-46bc-b1f4-468b9a2dfc0d"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
+PythonCall = "6099a3de-0909-46bc-b1f4-468b9a2dfc0d"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+TaijaData = "9d524318-b4e6-4a65-86d2-b2b72d07866c"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
diff --git a/test/runtests.jl b/test/runtests.jl
index 70ca7e6..5113264 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -1,10 +1,10 @@
 using CounterfactualExplanations
-using CounterfactualExplanations.Data
 using CounterfactualExplanations.DataPreprocessing
 using CounterfactualExplanations.Models
 using Printf
 using PythonCall
 using Random
+using TaijaData
 using TaijaInteroperability
 using Test
 
diff --git a/test/utils.jl b/test/utils.jl
index b0c6a41..586908a 100644
--- a/test/utils.jl
+++ b/test/utils.jl
@@ -6,8 +6,8 @@ Loads synthetic data, models, and generators.
 function _load_synthetic()
     # Data:
     data_sets = Dict(
-        :classification_binary => load_linearly_separable(),
-        :classification_multi => load_multi_class(),
+        :classification_binary => CounterfactualData(load_linearly_separable()...),
+        :classification_multi => CounterfactualData(load_multi_class()...),
     )
     # Models
     synthetic = Dict()