Skip to content

Commit eed107c

Browse files
committed
increase delay and number of retries when fetching newsgroups
1 parent 82f8d6a commit eed107c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
# -*- coding: utf-8 -*-
21
from os.path import dirname
32
from os.path import join
43
import csv
@@ -27,7 +26,7 @@ def _get_newsgroups(binary=False, remove_chrome=False, test=False, size=SIZE):
2726
subset = 'test' if test else 'train'
2827
data = fetch_20newsgroups(subset=subset, categories=categories,
2928
shuffle=True, random_state=42,
30-
remove=remove)
29+
remove=remove, n_retries=5, delay=5.0)
3130
assert data.target_names == categories
3231
return data.data[:size], data.target[:size], data.target_names
3332

@@ -51,6 +50,7 @@ def newsgroups_train_big():
5150
def newsgroups_train_binary_big():
5251
return _get_newsgroups(binary=True, remove_chrome=True, size=1000)
5352

53+
5454
class Bunch(dict):
5555
"""Container object for datasets: dictionary-like object that
5656
exposes its keys as attributes."""

0 commit comments

Comments
 (0)