dZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z d Z Gd d ejZd diZee_y)z"ai-platform local predict command.)absolute_import)division)unicode_literals)base)flags) local_utils)predict_utilities)logcR|jdddtjjj ||j d}|jdd|jd d |jd d tj j |y )z=Add arguments for `gcloud ai-platform local predict` command.z --model-dirTzPath to the model.)requiredhelp)r z--json-requesta, Path to a local file containing the body of JSON request. An example of a JSON request: { "instances": [ {"x": [1, 2], "y": [3, 4]}, {"x": [-1, -2], "y": [-3, -4]} ] } This flag accepts "-" for stdin. )r z--json-instancesa< Path to a local file from which instances are read. Instances are in JSON format; newline delimited. An example of the JSON instances file: {"images": [0.0, ..., 0.1], "key": 3} {"images": [0.0, ..., 0.1], "key": 2} ... This flag accepts "-" for stdin. z--text-instancesa& Path to a local file from which instances are read. Instances are in UTF-8 encoded text format; newline delimited. An example of the text instances file: 107,4.9,2.5,4.5,1.7 100,5.7,2.8,4.1,1.3 ... This flag accepts "-" for stdin. N) add_argumentrFRAMEWORK_MAPPER choice_arg AddToParseradd_mutually_exclusive_groupSIGNATURE_NAME)parsergroups (lib/surface/ai_platform/local/predict.py_AddLocalPredictArgsrsmd9MN##//7  - -t - <%        ""6*c&eZdZdZedZdZy)PredictzRun prediction locally.ct|y)N)r)rs rArgsz Predict.ArgsSs  rc"tjj|j}|r|jj nd}|j tjjdtj|j|j|j|j||j }|j!ds>t#|t$r|}n|j'd}t)j*||_|S)N tensorflowzIf the signature defined in the model is not serving_default then you must specify it via --signature-name flag, otherwise the command may fail.) json_requestjson_instancestext_instances frameworksignature_nameformat predictions)rrGetEnumForChoicer"namelowerr#r statusPrintr RunPredict model_dirrr r! IsSpecified isinstancelistgetr GetDefaultFormatr$)selfargsr"framework_flagresultsr%s rRunz Predict.RunWs&&77GI/8Y^^))+lN " jjPQ$$ &&**** ** ,G   H % GT " kk-0 %66{Cdk NrN)__name__ __module__ __qualname____doc__ staticmethodrr6rrrrPs!!rr DESCRIPTIONa'*{command}* performs prediction locally with the given instances. It requires the [TensorFlow SDK](https://www.tensorflow.org/install) be installed locally. The output format mirrors `gcloud ai-platform predict` (online prediction). You cannot use this command with custom prediction routines. N)r: __future__rrrgooglecloudsdk.callioper$googlecloudsdk.command_lib.ml_enginerrr googlecloudsdk.corer rCommandr_DETAILED_HELP detailed_helpr<rrrEsT)&'(6<B#1+hdllD  'r