fix: improve tolerance for error handling (#1991)

This commit is contained in:
Markus Cozowicz 2023-06-28 13:13:30 +02:00 коммит произвёл GitHub
Родитель bf2f4295bd
Коммит 0539752581
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
2 изменённых файлов: 3 добавлений и 4 удалений

Просмотреть файл

@ -166,7 +166,7 @@ object MADUtils extends Logging {
val modelStatus = modelInfo("status").asInstanceOf[JsString].value.toLowerCase
modelStatus match {
case "failed" =>
val errors = modelInfo("errors").convertTo[Seq[DMAError]].toJson.compactPrint
val errors = modelInfo("errors").toJson.compactPrint
throw new RuntimeException(s"Caught errors during fitting: $errors")
case "created" | "running" =>
throw new RuntimeException(s"model $modelId is not ready yet")
@ -258,7 +258,6 @@ trait MADHttpRequest extends HasURL with HasSubscriptionKey with HasAsyncReply {
}
} else {
val error = IOUtils.toString(response.entity.get.content, "UTF-8")
.parseJson.convertTo[DMAError].toJson.compactPrint
throw new RuntimeException(s"Caught error: $error")
}
}
@ -514,7 +513,7 @@ class SimpleFitMultivariateAnomaly(override val uid: String) extends Estimator[S
val modelId = response("modelId").convertTo[String]
if (modelInfo("status").asInstanceOf[JsString].value.toLowerCase() == "failed") {
val errors = modelInfo("errors").convertTo[Seq[DMAError]].toJson.compactPrint
val errors = modelInfo("errors").toJson.compactPrint
throw new RuntimeException(s"Caught errors during fitting: $errors")
}

Просмотреть файл

@ -28,7 +28,7 @@ case class DMASummary(status: String,
object DMAError extends SparkBindings[DMAError]
case class DMAError(code: String, message: String)
case class DMAError(code: Option[String], message: Option[String])
case class DMAVariableState(variable: Option[String],
filledNARatio: Option[Double],