Results 2 issues of msdhina

I've used "lambda x: jnp.mean(x[x > 0])" in place of "jnp.mean" `media_scaler = preprocessing.CustomScaler(divide_operation=lambda x: jnp.mean(x[x > 0])) ` It throws "NonConcreteBooleanIndexError: Array boolean indices must be concrete; got ShapedArray(bool[82])"...

How to get contribution for the extra features that is fed into the model?