get_img_fit_flow fit_smpl_size not accepting floats

[norahik]

hello!
i'm trying to use the function "get_img_fit_flow"
when i set 'fit_smpl_size' a float number (0.2)
the following line:
idx = np.random.choice(imgs.shape[0], batch_size*fit_smpl_size, replace=False)
raises an error:
"TypeError: 'float' object cannot be interpreted as an index"

i understand it cannot be an it since we done want to get all the data. could you please explain if i'm missing something?