code: def build_model(self, max_depth, num_round):
      dtrain, _ = tm.load_agaricus(__file__)
      param = {
          "max_depth": max_depth,
          "objective": "binary:logistic",
          "verbosity": 1,
      }
      num_round = num_round
      bst = xgb.train(param, dtrain, num_round)
      return bst
  
file_path: xgboost/tests/python/test_parse_tree.py
imports: import numpy
  import pytest
  import xgboost
  from xgboost import testing
label_desc: Use image node_id 1 for calling the TestTreesToDataFrame obj's underlying member method code with example usage: obj.build_model(max_depth, num_round) and returns: bst