dZddlmZddlmZddlmZddlmZddlZddlZddlZdZ dZ d Z d Z d Z d Zd ZdZdZedk(reyy)zUtilities for running predictions locally. This module will always be run within a subprocess, and therefore normal conventions of Cloud SDK do not apply here. )absolute_import)division)print_function)unicode_literalsNc:t|dtji|y)zPrint to stderr.fileN)printsysstderr)argskwargss 9lib/googlecloudsdk/command_lib/ml_engine/local_predict.pyeprintrs)CJJ)&)zsPlease verify the installed tensorflow version with: "python -c 'import tensorflow; print tensorflow.__version__'".zjPlease verify the installed sklearn version with: "python -c 'import sklearn; print sklearn.__version__'".zjPlease verify the installed xgboost version with: "python -c 'import xgboost; print xgboost.__version__'".c  ddlmcm} |j |kr t dj |ty y#t$rt dYywxYw#ttf$r}t d|dtYd}~yd}~wwxYw) z@Check whether TensorFlow is installed at an appropriate version.rNzNCannot import Tensorflow. Please verify "python -c 'import tensorflow'" works.Fz(Tensorflow version must be at least {} .z6Error while getting the installed TensorFlow version:  T) tensorflow.compat.v1compatv1 ImportErrorr __version__formatVERIFY_TENSORFLOW_VERSION NameErrorAttributeError)versiontfes r_verify_tensorflowr3s%%  ~~ 7 > >w G &(      67  ^ $ CQ *, s& =.AAAB&A==Bc> ddl} ddl} |j|kr tdj |t y y #t$rtdYywxYw#t$rtdYywxYw#ttf$r}td|dt Yd}~yd}~wwxYw) zBCheck whether scikit-learn is installed at an appropriate version.rNzfCannot import scipy, which is needed for scikit-learn. Please verify "python -c 'import scipy'" works.FzHCannot import sklearn. Please verify "python -c 'import sklearn'" works.z*Scikit-learn version must be at least {} .z8Error while getting the installed scikit-learn version: rT) scipyrrsklearnrrVERIFY_SCIKIT_LEARN_VERSIONrr)rr!r"rs r_verify_scikit_learnr$Ks  W$ 9 @ @ I (* % -   89    34  ^ $ Eq$ &( s7=A.A1AAA.-A.1BBBc ddl} |j|kr tdj |t y y#t$rtdYywxYw#t tf$r}td|dt Yd}~yd}~wwxYw) z=Check whether xgboost is installed at an appropriate version.rNzHCannot import xgboost. Please verify "python -c 'import xgboost'" works.Fz%Xgboost version must be at least {} .z3Error while getting the installed xgboost version: rT)xgboostrrrrVERIFY_XGBOOST_VERSIONrr)rr&rs r_verify_xgboostr(ks W$ 4 ; ;G D #% %    34  ^ $ @!T !# s&8.AAAA=!A88A=c|dk(r!tdstjdy|dk(r!tdstjdy|dk(r"t dstjdyyy) z=Verifies the appropriate ML libs are installed per framework. tensorflowz1.0.0 scikit_learnz0.18.1r&z0.6a2N)rr exitr$r() frameworks r_verify_ml_libsr/sZ,'9''BHHRLN"+?+IHHRLIog&>HHRL'?rctj}|jddd|jdddd|jd dd |j\}}|j"d d lm}|j|j}n |j}|r t|d d lm }g}tjD]7}tj|jd}|j!|9|j#|j|||j$} t'tj(| y)Nz --model-dirTzPath of the model.)requiredhelpz --frameworkFzThe ML framework used to train this version of the model. If not specified, the framework will be identified based on the model file name stored in the specified model-dir)r1defaultr2z--signature-namez0Tensorflow signature to select input/output map.r)prediction_utils)prediction_libr) model_dir instancesr.signature_name)argparseArgumentParser add_argumentparse_known_argsr.cloud.ml.predictionr4detect_frameworkr6r/r5r stdinjsonloadsrstripappend local_predictr8r dumps) parserr _r4r.r5r7lineinstance predictionss rmainrKs-  " " $&md9MN E G (5MO  # # %'$ ^^4 11$..AIII 1)iidzz$++d+,H X,,(( -*+  ; r__main__)__doc__ __future__rrrrr9r@r rrr#r'rr$r(r/rK__name__rrrQsn '%' *@ ?: 0@0'!T z&r