Fix parsing of different exception string formats (#4785)

diff --git a/python/tvm/_ffi/base.py b/python/tvm/_ffi/base.py
index cdcc131..36effa3 100644 (file)
--- a/python/tvm/_ffi/base.py
+++ b/python/tvm/_ffi/base.py
@@ -205,8 +205,9 @@ def _find_error_type(line):
             return None
         start_pos = line.rfind(":", 0, end_pos)
         if start_pos == -1:
-            return None
-        err_name = line[start_pos + 1 : end_pos].strip()
+            err_name = line[:end_pos].strip()
+        else:
+            err_name = line[start_pos + 1 : end_pos].strip()
         if _valid_error_name(err_name):
             return err_name
         return None