Completed
Pull Request — master (#297)
by Bart
01:19
created

fuel.datasets.YouTubeAudio.__init__()   A

Complexity

Conditions 1

Size

Total Lines 4

Duplication

Lines 0
Ratio 0 %
Metric Value
cc 1
dl 0
loc 4
rs 10
1
from fuel.datasets.hdf5 import H5PYDataset
2
from fuel.utils import find_in_data_path
3
4
5
class YouTubeAudio(H5PYDataset):
6
    def __init__(self, youtube_id, **kwargs):
7
        super(YouTubeAudio, self).__init__(
8
            file_or_path=find_in_data_path('{}.hdf5'.format(youtube_id)),
9
            which_sets=('train',), **kwargs
10
        )
11