File size: 1,018 Bytes
abd2a81 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# class Map(object):
# """Applies parameter transform to all items of input list"""
#
# def __init__(self, transform, multi_outs=False):
# self.transform = transform
# if multi_outs:
# self.__call__ = self._map_multi_outs
# else:
# self.__call__ = self._map_single_out
#
# def _map_single_out(self, data_list):
# assert type(data_list) == list, "data_list should be a list"
# return [self.transform(item) for item in data_list]
#
# def _map_multi_outs(self, data_list):
# assert type(data_list) == list, "data_list should be a list"
# return tuple(zip(*[self.transform(item) for item in data_list]))
class Map(object):
"""Applies parameter transform to all items of input list"""
def __init__(self, transform):
self.transform = transform
def __call__(self, data_list):
assert type(data_list) == list, "data_list should be a list"
return [self.transform(item) for item in data_list]
|