Remove DST prefix

Author: juliohm

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

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