def METHOD_NAME(self, model_fn, **kwargs):
    if self._hvd:
        from tensorflow.python.distribute.hvd_strategy import wraps_estimator
        _estimator = wraps_estimator(_estimator_lib.Estimator)
    elif self._hb:
        _estimator = hb.METHOD_NAME.Estimator
    return _estimator(model_fn, **kwargs)