FIX: use the mldata mock in docstring as well