From 34144fc13a4f12e83301fbebd581466cb83eb2c2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?J=C3=BAlio=20Hoffimann?= Date: Fri, 18 Oct 2024 13:27:53 -0300 Subject: [PATCH] Remove DST prefix --- src/StatsLearnModels.jl | 1 - src/models/nearestneighbors.jl | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/StatsLearnModels.jl b/src/StatsLearnModels.jl index 2194db3..c9a1ea7 100644 --- a/src/StatsLearnModels.jl +++ b/src/StatsLearnModels.jl @@ -11,7 +11,6 @@ using StatsBase: mode, mean using ColumnSelectors: ColumnSelector, selector using TableTransforms: StatelessFeatureTransform -import DataScienceTraits as DST import TableTransforms: applyfeat, isrevertible using DecisionTree: AdaBoostStumpClassifier, DecisionTreeClassifier, RandomForestClassifier diff --git a/src/models/nearestneighbors.jl b/src/models/nearestneighbors.jl index 4d0d45d..7b5a168 100644 --- a/src/models/nearestneighbors.jl +++ b/src/models/nearestneighbors.jl @@ -48,13 +48,13 @@ function predict(fmodel::FittedModel{<:NearestNeighborsModel}, table) end function _checkoutput(::KNNClassifier, x) - if !(elscitype(x) <: DST.Categorical) + if !(elscitype(x) <: Categorical) throw(ArgumentError("output column must be categorical")) end end function _checkoutput(::KNNRegressor, x) - if !(elscitype(x) <: DST.Continuous) + if !(elscitype(x) <: Continuous) throw(ArgumentError("output column must be continuous")) end end