Skip to content
Snippets Groups Projects
Commit 237370ba authored by Mathieu Blondel's avatar Mathieu Blondel
Browse files

Pass kwargs to mlcomp loader.

parent b18b9065
No related branches found
No related tags found
No related merge requests found
......@@ -7,10 +7,10 @@ import numpy as np
from scikits.learn.datasets.base import load_files
def _load_document_classification(dataset_path, metadata, set_=None):
def _load_document_classification(dataset_path, metadata, set_=None, **kwargs):
if set_ is not None:
dataset_path = os.path.join(dataset_path, set_)
return load_files(dataset_path, metadata.get('description'))
return load_files(dataset_path, metadata.get('description'), **kwargs)
LOADERS = {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment