o eoã@sddlmZGdd„dƒZdS)é)Ú raise_errorc@s<eZdZdZddd„Zdd„Zddd „Zd d „Zd d „ZdS)ÚInferRequestedOutputaòAn object of InferRequestedOutput class is used to describe a requested output tensor for an inference request. Parameters ---------- name : str The name of output tensor to associate with this object. binary_data : bool Indicates whether to return result data for the output in binary format or explicit tensor within JSON. The default value is True, which means the data will be delivered as binary data in the HTTP body after JSON object. This field will be unset if shared memory is set for the output. class_count : int The number of classifications to be requested. The default value is 0 which means the classification results are not requested. TrcCs2||_i|_|dkr||jd<||_||jd<dS)NrÚclassificationÚ binary_data)Ú_nameÚ _parametersÚ_binary)ÚselfÚnamerZ class_count©r úSD:\Projects\ConvertPro\env\Lib\site-packages\tritonclient/http/_requested_output.pyÚ__init__3s  zInferRequestedOutput.__init__cCs|jS)zˆGet the name of output associated with this object. Returns ------- str The name of output )r©r r r r r ;szInferRequestedOutput.namecCsPd|jvr tdƒ|jrd|jd<||jd<||jd<|dkr&||jd<d Sd S) aÚMarks the output to return the inference result in specified shared memory region. Parameters ---------- region_name : str The name of the shared memory region to hold tensor data. byte_size : int The size of the shared memory region to hold tensor data. offset : int The offset, in bytes, into the region where the data for the tensor starts. The default value is 0. rz3shared memory can't be set on classification outputFrÚshared_memory_regionÚshared_memory_byte_sizerÚshared_memory_offsetN)rrr)r Z region_nameZ byte_sizeÚoffsetr r r Úset_shared_memoryEs    ÿz&InferRequestedOutput.set_shared_memorycCs:|j|jd<|j dd¡|j dd¡|j dd¡dS)zåClears the shared memory option set by the last call to InferRequestedOutput.set_shared_memory(). After call to this function requested output will no longer be returned in a shared memory region. rrNrr)rrÚpoprr r r Úunset_shared_memory^s z(InferRequestedOutput.unset_shared_memorycCsd|ji}|jr |j|d<|S)zŽRetrieve the underlying input as json dict. Returns ------- dict The underlying tensor as a dict r Ú parameters)rr)r Ztensorr r r Ú _get_tensorjs  z InferRequestedOutput._get_tensorN)Tr)r) Ú__name__Ú __module__Ú __qualname__Ú__doc__r r rrrr r r r rs   rN)Ztritonclient.utilsrrr r r r Ús