diff --git a/Project.toml b/Project.toml
index 4a2ddda..da28fc0 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "GeoStatsValidation"
 uuid = "36f43c0d-3673-45fc-9557-6860e708e7aa"
 authors = ["Elias Carvalho <eliascarvdev@gmail.com> and contributors"]
-version = "0.3.16"
+version = "0.3.17"
 
 [deps]
 ColumnSelectors = "9cc86067-7e36-4c61-b350-1ac9833d277f"
@@ -17,7 +17,7 @@ StatsLearnModels = "c146b59d-1589-421c-8e09-a22e554fd05c"
 
 [compat]
 ColumnSelectors = "1.0"
-DataScienceTraits = "0.4"
+DataScienceTraits = "1.0"
 DensityRatioEstimation = "1.2"
 GeoStatsBase = "0.46"
 GeoStatsModels = "0.6"
diff --git a/src/GeoStatsValidation.jl b/src/GeoStatsValidation.jl
index 284cbf2..f99c15f 100644
--- a/src/GeoStatsValidation.jl
+++ b/src/GeoStatsValidation.jl
@@ -18,7 +18,6 @@ using ColumnSelectors: selector
 using GeoStatsBase: weight, folds, mean
 using GeoStatsBase: WeightingMethod, DensityRatioWeighting, UniformWeighting
 using GeoStatsBase: FoldingMethod, BallFolding, BlockFolding, OneFolding, UniformFolding
-using DataScienceTraits: Continuous, Categorical
 using LossFunctions: L2DistLoss, MisclassLoss
 using LossFunctions.Traits: SupervisedLoss