nseunghee97

I tried the code below by referring to the [documentation](https://jax.readthedocs.io/en/latest/errors.html#jax.errors.NonConcreteBooleanIndexError) @michevan suggested. ``` def nonzero_mean(arr): return jnp.nanmean(jnp.where(arr != 0, arr, jnp.nan)) media_scaler = preprocessing.CustomScaler(divide_operation=nonzero_mean) ``` I first converted zero values...