Skip to content

Commit 6e2da25

Browse files
committed
returning to default implementation for onehot because I don't want to break it.
1 parent fcda965 commit 6e2da25

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/onehot.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ end
220220

221221
@nograd OneHotArray, onecold, onehot, onehotbatch
222222

223-
function Base.:(*)(A::AbstractMatrix, B::OneHotLike{<:Any, L, 0}) where L
223+
function Base.:(*)(A::AbstractMatrix, B::OneHotLike{<:Any, L}) where L
224224
_isonehot(B) || return invoke(*, Tuple{AbstractMatrix, AbstractMatrix}, A, B)
225225
size(A, 2) == L || throw(DimensionMismatch("Matrix column must correspond with OneHot size: $(size(A, 2)) != $L"))
226226
return A[:, onecold(B)]

0 commit comments

Comments
 (0)