Could you pass the scale_factor as an integer or are you relying on the float value?
CC @yf225 the docs seem to miss the recompute_scale_factor argument from the last PR 4 (maybe I’m missing another WIP PR  ).