fix python 2 compatibility iterating over a pandas Series

oesteban · oesteban · commit e6d2b805a40c · 2017-03-23T13:56:29.000-07:00
diff --git a/nipype/utils/draw_gantt_chart.py b/nipype/utils/draw_gantt_chart.py
@@ -364,8 +364,13 @@ def draw_resource_bar(start_time, finish_time, time_series, space_between_minute
     space_between_minutes = space_between_minutes / scale
 
     # Iterate through time series
+    if PY3:
+        ts_items = time_series.items()
+    else:
+        ts_items = time_series.iteritems()
+
     ts_len = len(time_series)
-    for idx, (ts_start, amount) in enumerate(time_series.items()):
+    for idx, (ts_start, amount) in enumerate(ts_items):
         if idx < ts_len-1:
             ts_end = time_series.index[idx+1]
         else:
