move xgboost import

  if hyper_parameters is None:

    hyper_parameters = hps[model]

  model_name = model

  import xgboost

  if model_name == 'tf':

    # Loading hyper parameters

    # Building xgboost classification model

    def model_builder(model_dir_xgb):

      import xgboost

      xgboost_model = xgboost.XGBClassifier(

          max_depth=max_depth,

          learning_rate=learning_rate,