ValueError: The parameter loc has invalid values

I am trying Ray tune wrapper for Optuna to tune my Stable Baslines3 model. But I am getting this error ValueError: The parameter loc has invalid values. This occurs probably when we divide by zero, that is when the model diverges. In Optuna there is a catch parameter that ignores value error and tries out the next set of hyperparameters. It is present in the optimised attribute of the study created. But I could not find the source code for the wrapper Source code for Wrapper. Is there a similar way to do this in ray tune? How can I handle this exception? Thanks in advance