Fix exeption handling of unpacking k/v pair.

douban · Jan 23, 2019 · 17e83f6 · 17e83f6
1 parent 6dfe760
commit 17e83f6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dpark/task.py b/dpark/task.py
@@ -214,8 +214,9 @@ def _run(self, task_id):
             try:
                 try:
                     k, v = item
-                except:
-                    raise DparkUserFatalError("item of {} should be (k, v) pair, got: {}".format(rdd.scope.key, item))
+                except (TypeError, ValueError) as e:
+                    msg = "item of {} should be (k, v) pair, got: {}, exception: {}".format(rdd.scope.key, item, e)
+                    raise DparkUserFatalError(msg)
 
                 bucket = buckets[get_partition(k)]
                 r = bucket.get(k, None)