U UºcŽã@sBdZddlmZddlmZGdd„deƒZdd„Zd d d „ZdS) z,Common utils for all inference plugin files.é)Úclassification_pb2)Úregression_pb2c@s$eZdZdZdd„Zedd„ƒZdS)ÚInvalidUserInputErrorz¬An exception to throw if user input is detected to be invalid. Attributes: original_exception: The triggering `Exception` object to be wrapped, or a string. cCs||_t |¡dS)zInits InvalidUserInputError.N)Úoriginal_exceptionÚ ExceptionÚ__init__)Úselfr©r úN/tmp/pip-unpacked-wheel-15h0ro02/tensorboard_plugin_wit/_utils/common_utils.pyrszInvalidUserInputError.__init__cCsdt|jƒS)NzInvalidUserInputError: )Ústrr)rr r r Úmessage"szInvalidUserInputError.messageN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚpropertyr r r r r rsrcCs~|j|j}|j}|jdkrlg}t|jjdjƒD]6}||jjdj}| ||||jjdj…¡q2n|}t |||j ƒS)aLConverts a PredictResponse to ClassificationResponse or RegressionResponse. Args: pred: PredictResponse to convert. serving_bundle: A `ServingBundle` object that contains the information about the serving request that the response was generated by. Returns: A ClassificationResponse or RegressionResponse. Úclassificationré) ÚoutputsZpredict_output_tensorZ float_valÚ model_typeÚrangeZ tensor_shapeZdimÚsizeÚappendÚconvert_prediction_valuesÚ model_spec)ÚpredÚserving_bundleÚoutputZ raw_outputÚvaluesÚ example_indexÚstartr r r Úconvert_predict_response's  $r!Nc Cs®|jdkrjt ¡}tt|ƒƒD]H}|jj ¡}tt||ƒƒD]&}|j ¡}||||_ t |ƒ|_ q>qn0t   ¡}tt|ƒƒD]}|jj ¡}|||_q~|rª|j |¡|S)a“Converts tensor values into ClassificationResponse or RegressionResponse. Args: values: For classification, a 2D list of numbers. The first dimension is for each example being predicted. The second dimension are the probabilities for each class ID in the prediction. For regression, a 1D list of numbers, with a regression score for each example being predicted. serving_bundle: A `ServingBundle` object that contains the information about the serving request that the response was generated by. model_spec: Optional model spec to put into the response. Returns: A ClassificationResponse or RegressionResponse. r)rrZClassificationResponserÚlenÚresultZclassificationsÚaddÚclassesZscorer ÚlabelrZRegressionResponseZ regressionsÚvaluerZCopyFrom) rrrÚresponserrZ class_indexZ class_scoreZ regressionr r r r=s      r)N)rZ6tensorboard_plugin_wit._vendor.tensorflow_serving.apisrrrrr!rr r r r Ús