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

时间:2017-02-21 23:41:22

标签: google-cloud-ml

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

google/cloud/ml/dataflow/batch_prediction.py的第153-159行,我们有以下代码:

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))

但是,异常对象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)']

我认为不应该引发这些错误,因为except块的重点是捕获这些错误并将它们作为副输出传递给ParDo。

1 个答案:

答案 0 :(得分:2)

感谢报道此事!我们已经修复了它,很快就会在下一个版本中提供一个好的版本。