deep-spin / entmax

The entmax mapping and its loss, a family of sparse softmax alternatives.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

EntmaxBisect forward requires unused argument

bpopeters opened this issue · comments

The forward method EntmaxBisect takes a required alpha argument. However, what is actually passed passed to the entmax_bisect function is self.alpha.

Good catch. That argument from the forward method should be removed, IMO.

This is fixed in code but not in the default package when installed. I think it would be worth a 1.0.1 release.

Edit: For clarity, if you install the current package and try to use EntmaxBisect class, it fails when forward is called. The existing code change fixes that. A 1.0.1 release would ensure the default package contains this fix.