code: def test_with_shap() -> None:
      from sklearn.datasets import fetch_california_housing
  
      X, y = fetch_california_housing(return_X_y=True)
      dtrain = xgb.DMatrix(X, label=y)
      model = xgb.train({"learning_rate": 0.01}, dtrain, 10)
      explainer = shap.TreeExplainer(model)
      shap_values = explainer.shap_values(X)
      margin = model.predict(dtrain, output_margin=True)
      assert np.allclose(
          np.sum(shap_values, axis=len(shap_values.shape) - 1),
          margin - explainer.expected_value,
          1e-3,
          1e-3,
      )
  
imports: import numpy
  import pytest
  import xgboost
label_desc: Use image node_id 1 for calling a global function with example usage: test_with_shap() without return types