Wondering how to make this work with integer valued parameter

[grechasneak]

Hey love the implementation it is very clean.

I am having some issues making it work for integer-valued hyperparameters and am wondering if you have any suggestions?