Update compat

Author: juliohm

Project.toml

@@ -1,7 +1,7 @@
 name = "TableTransforms"
 uuid = "0d432bfd-3ee1-4ac1-886a-39f05cc69a3e"
 authors = ["Júlio Hoffimann <[email protected]> and contributors"]
-version = "1.36.0"
+version = "1.36.1"
 
 [deps]
 AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
@@ -26,7 +26,7 @@ Unitful = "1986cc42-f94f-5a68-af5c-568840ba703d"
 
 [compat]
 AbstractTrees = "0.4"
-CategoricalArrays = "0.10"
+CategoricalArrays = "1.0"
 CoDa = "1.2"
 ColumnSelectors = "1.0"
 DataScienceTraits = "1.0"

src/transforms/onehot.jl

@@ -30,21 +30,14 @@ assertions(transform::OneHot) = [scitypeassert(Categorical, transform.selector)]
 
 isrevertible(::Type{<:OneHot}) = true
 
-_categ(x) = categorical(x), identity
-function _categ(x::CategoricalArray)
-  l, o = levels(x), isordered(x)
-  revfun = y -> categorical(y, levels=l, ordered=o)
-  x, revfun
-end
-
 function applyfeat(transform::OneHot, feat, prep)
   cols = Tables.columns(feat)
   names = Tables.columnnames(cols) |> collect
   columns = Any[Tables.getcolumn(cols, nm) for nm in names]
 
   name = selectsingle(transform.selector, names)
   ind = findfirst(==(name), names)
-  x, revfun = _categ(columns[ind])
+  x, revfun = _categorical(columns[ind])
 
   xlevels = levels(x)
   onehot = map(xlevels) do l
@@ -89,3 +82,10 @@ function revertfeat(::OneHot, newfeat, fcache)
   𝒯 = (; zip(names, columns)...)
   𝒯 |> Tables.materializer(newfeat)
 end
+
+_categorical(x) = x, identity
+function _categorical(x::CategoricalArray)
+  l, o = levels(x), isordered(x)
+  revfun = y -> categorical(y, levels=l, ordered=o)
+  x, revfun
+end