:dZddlmZddlmZddZdZ d dZy) zUtilities for running predictions. Includes (from the Cloud ML SDK): - _predict_lib Important changes: - Remove interfaces for TensorFlowModel (they don't change behavior). - Set from_client(skip_preprocessing=True) and remove the pre-processing code. )custom_code_utils)prediction_utilsNc Ptj|d}|r|S|xstj}|tjk(rddlm}|j }nK|tjk(rddlm}|j}n%|tjk(rddlm}|j}|S)aCreates and returns the appropriate model. Creates and returns a Model if no user specified model is provided. Otherwise, the user specified model is imported, created, and returned. Args: client: An instance of PredictionClient for performing prediction. model_path: The path to the exported model (e.g. session_bundle or SavedModel) framework: The framework used to train the model. Returns: An instance of the appropriate model class. Nrtf_prediction_libsk_xg_prediction_lib) rcreate_user_modelrTENSORFLOW_FRAMEWORK_NAME frameworksrTensorFlowModelSCIKIT_LEARN_FRAMEWORK_NAMEr SklearnModelXGBOOST_FRAMEWORK_NAME XGBoostModel)client model_path framework unused_kwargs custom_modelr model_clsr s r0s,  F0:BF  ,r