ssh_wrapper = SSHEstimatorWrapper.create(estimator, connection_wait_time_seconds=0)
Seems passing this does not take effect, I need to manually stop sm-wait even though the wait time is 0. I was able to resolve it by doing:
ssh_wrapper = SSHEstimatorWrapper.create(estimator, connection_wait_time_seconds=0, connection_wait_time=timedelta(seconds=0))
Seems like either there is a bug or the docs need to be updated