fix python3 issue

intel · Feb 12, 2018 · 68ca869 · 68ca869
1 parent 4696b8f
commit 68ca869
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/ideep4py/__init__.py b/python/ideep4py/__init__.py
@@ -116,8 +116,10 @@ def all_ready(inputs, supported_ndim=(2, 4)):
 def split(x, indices_or_sections, axis=0):
     if all_ready((x,)):
         offsets = intVector()
+        # FIXME
+        # bypass python3 issue
         for i in indices_or_sections:
-            offsets.push_back(i)
+            offsets.push_back(int(i))
         ys = concat.Backward(x, offsets, axis)
 
         if ys: