o c²“iò ã@s€ddlZddlmZmZmZmZddlZddlZ ddl m Z m Z m Z ddlmZddlmZedƒZedƒZGdd „d eƒZdS) éN)ÚDictÚOptionalÚTypeVarÚUnion)Ú BatchFormatÚ_convert_batch_type_to_pandasÚ_convert_pandas_to_batch_type)Ú Predictor)Ú DeveloperAPIÚ TensorTypeÚ TensorDtypec @sLeZdZejdeejee ejffde ee ee e ffdee ee e fffdd„ƒZ ejde dejfdd„ƒZejed ee ee e ffdee ee e fffd d „ƒƒZeedefd d „ƒƒZdejde ee ee e ffdejfdd„Zdeejee ejffde ee ee e ffdeejee ejfffdd„ZdS)Ú DLPredictorÚ numpy_arraysÚdtypeÚreturncCót‚)aIConverts a NumPy ndarray batch to the tensor type for the DL framework. Args: numpy_array: The numpy array to convert to a tensor. dtype: The tensor dtype to use when creating the DL tensor. ndarray: A (dict of) NumPy ndarray(s) that we wish to convert to a (dict of) tensor(s). dtype: A (dict of) tensor dtype(s) to use when creating the DL tensor; if None, the dtype will be inferred from the NumPy ndarray data. Returns: A deep learning framework specific tensor. ©ÚNotImplementedError)Úselfrr©rúT/home/ubuntu/.local/lib/python3.10/site-packages/ray/train/_internal/dl_predictor.pyÚ_arrays_to_tensorsszDLPredictor._arrays_to_tensorsÚtensorcCr)zÏConverts tensor framework specific tensor to a numpy array. Args: tensor: A framework specific tensor. Returns: A numpy array representing the input tensor. r)rrrrrÚ_tensor_to_array)s zDLPredictor._tensor_to_arrayÚinputscCr)zôInputs the tensor to the model for this Predictor and returns the result. Args: inputs: The tensor to input to the model. Returns: A tensor or dictionary of tensors containing the model output. r)rrrrrÚ call_model6s zDLPredictor.call_modelcCstjS©N)rÚNUMPY)ÚclsrrrÚpreferred_batch_formatEsz"DLPredictor.preferred_batch_formatÚdatacCs$t|tj|jƒ}| ||¡}t|ƒSr)rrrÚ_cast_tensor_columnsÚ_predict_numpyr)rr rÚ numpy_inputÚ numpy_outputrrrÚ_predict_pandasJsý zDLPredictor._predict_pandascsjt|tƒrt|ƒdkrtt| ¡ƒƒ}ˆ ||¡}ˆ |¡}t|tƒr.‡fdd„| ¡DƒSdˆ  |¡iS)Nécsi|] \}}|ˆ |¡“qSr)r)Ú.0ÚkÚv©rrrÚ esz.DLPredictor._predict_numpy..Ú predictions) Ú isinstanceÚdictÚlenÚnextÚiterÚvaluesrrÚitemsr)rr rÚ model_inputÚ model_outputrr*rr"Ws   zDLPredictor._predict_numpyN)Ú__name__Ú __module__Ú __qualname__ÚabcÚabstractmethodrÚnpÚndarrayrÚstrrr r rrr rÚ classmethodrrÚpdÚ DataFramer%r"rrrrr sFþý ü ÿþ þý ü þýür )r9ÚtypingrrrrÚnumpyr;Úpandasr?Ú"ray.air.util.data_batch_conversionrrrÚray.train.predictorr Úray.util.annotationsr r r r rrrrÚs