Skip to content

Commit 2ae707d

Browse files
committed
Refactor indicator.jl
1 parent d1b5151 commit 2ae707d

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

src/transforms/indicator.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
1-
const SCALES = [:quantile, :linear]
1+
# ------------------------------------------------------------------
2+
# Licensed under the MIT License. See LICENSE in the project root.
3+
# ------------------------------------------------------------------
24

35
"""
46
Indicator(col; k=10, scale=:quantile, categ=false)
@@ -34,7 +36,7 @@ struct Indicator{S<:SingleColumnSelector} <: StatelessFeatureTransform
3436
throw(ArgumentError("`k` must be greater than or equal to 1"))
3537
end
3638

37-
if scale SCALES
39+
if scale (:quantile, :linear)
3840
throw(ArgumentError("invalid `scale` option, use `:quantile` or `:linear`"))
3941
end
4042

0 commit comments

Comments
 (0)