fix ci

apache · Jun 9, 2020 · 58103c9 · 58103c9
1 parent 6c4bb92
commit 58103c9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/tvm/relay/op/transform.py b/python/tvm/relay/op/transform.py
@@ -217,10 +217,11 @@ def reshape(data, newshape):
         for shape in newshape:
             if isinstance(shape, IntImm):
                 tempshape.append(shape.value)
-            elif isinstance(shape, int):
-                tempshape.append(shape)
             else:
-                raise RuntimeError('Unrecognized shape type: %s' % type(shape))
+                try:
+                    tempshape.append(int(shape))
+                except ValueError as err:
+                    raise RuntimeError('Unrecognized shape type: %s' % err)
         newshape = const(tempshape)
     return _make.reshape(data, newshape)