google-cloud-ml - google.cloud.ml.dataflow.batch_prediction.PredictionDoFn 中的错误处理逻辑错误

Question

我在看最新版本 ( obtained from gs://cloud-ml/sdk/cloudml-0.1.9-alpha.dataflow.tar.gz)

在第 153-159 行google/cloud/ml/dataflow/batch_prediction.py，我们有以下代码：

except Exception as e:  # pylint: disable=broad-except
  logging.error("Got an unknown exception: [%s].", traceback.format_exc())
  if self._cloud_logger:
    self._cloud_logger.write_error_message(
        str(e), self._create_snippet(context.element))
  yield beam.pvalue.SideOutputValue("errors",
                                    (e.error_message, context.element))

但是，Exception 对象e可能没有属性error_message。这有时会给我如下错误：

• AttributeError: 'exceptions.RuntimeError' object has no attribute 'error_message' [while running 'Evaluate/ParDo(PredictionDoFn)']
• AttributeError: 'exceptions.KeyError' object has no attribute 'error_message' [while running 'Evaluate/ParDo(PredictionDoFn)']
• ETC

我认为不应引发这些错误，因为 except 块的全部意义在于捕获这些错误并将它们作为辅助输出传递到 ParDo 之外。

Answer 1

感谢您报告此事！我们已经修复了它，并且很快就会在下一个版本中提供一个好的版本。