convert: benchmark: Make converter benchmark Py3k compatible

1 files changed, 5 insertions, 4 deletions

diff --git a/host/utils/converter_benchmark.py b/host/utils/converter_benchmark.py
index c24c1f983..5e6cba11c 100644
--- a/host/utils/converter_benchmark.py
+++ b/host/utils/converter_benchmark.py
@@ -13,6 +13,7 @@ from __future__ import print_function
 import argparse
 import csv
 import subprocess
+from six import iteritems
 
 INTRO_SETUP = {
     'n_samples': {
@@ -38,7 +39,7 @@ TABLE_SETUP = {
 def run_benchmark(args):
     """ Run the tool with the given arguments, return the section in the {{{ }}} brackets """
     call_args = ['./converter_benchmark',]
-    for k, v in args.__dict__.iteritems():
+    for k, v in iteritems(args.__dict__):
         k = k.replace('_', '-')
         if v is None:
             continue
@@ -50,7 +51,7 @@ def run_benchmark(args):
         call_args.append(str(v))
     print(call_args)
     try:
-        output = subprocess.check_output(call_args)
+        output = subprocess.check_output(call_args).decode('utf-8')
     except subprocess.CalledProcessError as ex:
         print(ex.output)
         exit(ex.returncode)
@@ -64,12 +65,12 @@ def print_stats_table(args, csv_output):
     Print stats.
     """
     reader = csv.reader(csv_output.strip().split('\n'), delimiter=',')
-    title_row = reader.next()
+    title_row = next(reader)
     row_widths = [0,] * len(TABLE_SETUP)
     for idx, row in enumerate(reader):
         if idx == 0:
             # Print intro:
-            for k, v in INTRO_SETUP.iteritems():
+            for k, v in iteritems(INTRO_SETUP):
                 print("{title}: {value}".format(
                     title=v['title'],
                     value=row[title_row.index(k)],