def build(self, config):
        from zoo.orca.automl.xgboost.XGBoost import XGBoost
        model = XGBoost(model_type=self.model_type, config=config)

        if self.n_cpus is not None:
            model.set_params(n_jobs=self.n_cpus)
        return model
 def create_model():
     _model = XGBoost(model_type=model_type, config=config)
     if "cpu" in resources_per_trial:
         _model.set_params(n_jobs=resources_per_trial.get("cpu"))
     return _model