FIX: doctests in partial_dependence