MAINT Update whitelisted estimators in test_docstrings.py
Add DecisionTreeClassifier & DecisionTreeRegressor to whitelist
@@ -24,8 +24,10 @@ DOCSTRING_WHITELIST = [
"RidgeClassifierCV.decision_function",
"SGDClassifier.decision_function",
"KernelDensity",
- "KernelDensity.fit"
- "KernelDensity.score"
+ "KernelDensity.fit",
+ "KernelDensity.score",
+ "DecisionTreeClassifier",
+ "DecisionTreeRegressor"
]
