# 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]