adding one more exception type for if work_result is None

SuperCowPowers · Jun 22, 2014 · ad100a5 · ad100a5
1 parent d887d81
commit ad100a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/workbench.py b/server/workbench.py
@@ -277,8 +277,8 @@ def work_request(self, worker_class, md5, subkeys=None):
                         tmp = tmp[key]
                     sub_results[key] = tmp
                 work_results = sub_results
-            except KeyError:
-                raise KeyError('Could not get one or more subkeys for: %s' % (work_results))
+            except (KeyError, TypeError):
+                raise RuntimeError('Could not get one or more subkeys for: %s' % (work_results))
 
         # Clean it and ship it
         work_results = self.data_store.clean_for_serialization(work_results)