diff --git a/bob/measure/load.py b/bob/measure/load.py index 0ef498c617bb40fa8afa3d4756b71e728fb860ba..d5996563c9d08078967e63bf9d9a70815585df1f 100644 --- a/bob/measure/load.py +++ b/bob/measure/load.py @@ -9,6 +9,7 @@ import numpy import csv import tarfile import os +import sys import logging logger = logging.getLogger('bob.measure') @@ -93,7 +94,12 @@ def four_column(filename): """ - reader = csv.reader(open_file(filename, mode='rb'), delimiter=' ') + opened = open_file(filename, 'rb') + if sys.version_info.major > 2: + import io + opened = io.TextIOWrapper(opened, newline="") + + reader = csv.reader(opened, delimiter=' ') for splits in reader: splits[-1] = float(splits[-1]) yield splits @@ -214,7 +220,12 @@ def five_column(filename): """ - reader = csv.reader(open_file(filename, mode='rb'), delimiter=' ') + opened = open_file(filename, 'rb') + if sys.version_info.major > 2: + import io + opened = io.TextIOWrapper(opened, newline="") + + reader = csv.reader(opened, delimiter=' ') for splits in reader: splits[-1] = float(splits[-1]) yield splits