graphviz>=0.9
pandas
numpy
scikit-learn
matplotlib
colour
pytest

[all]
xgboost
pyspark
lightgbm
tensorflow_decision_forests

[lightgbm]
lightgbm

[pyspark]
pyspark

[tensorflow_decision_forests]
tensorflow_decision_forests

[xgboost]
xgboost
