useDynLib(grf)

importFrom("Matrix", "Matrix", "cBind")
importFrom("methods", "new")
importFrom("stats", "coef", "lm", "predict", "runif", "var", "weighted.mean")
importFrom("utils", "capture.output")
import(Rcpp)

S3method(plot,grf_tree)
S3method(predict,causal_forest)
S3method(predict,custom_forest)
S3method(predict,instrumental_forest)
S3method(predict,local_linear_forest)
S3method(predict,quantile_forest)
S3method(predict,regression_forest)
S3method(print,grf)
S3method(print,grf_tree)

export(average_treatment_effect)
export(average_partial_effect)
export(causal_forest)
export(custom_forest)
export(get_sample_weights)
export(get_tree)
export(instrumental_forest)
export(local_linear_forest)
export(quantile_forest)
export(regression_forest)
export(split_frequencies)
export(test_calibration)
export(tune_causal_forest)
export(tune_local_linear_forest)
export(tune_regression_forest)
export(variable_importance)
