U /e@sddlmZddlZddlZddlZddlZddlmZeej edkZ ej Z e e ZddZde ddd Zd d d d ZdS)) annotationsN)parsez2.0.0cCsn|jjd\}}}|dkr0ddl}t||jS|dkrPddl}t||jjj S|dkrft|t t fSdSdS)N.numpyrpandasbuiltinsF) __class__ __module__ partitionr isinstanceZndarrayrcoreZgenericZNDFramestrbytes)xmod_nppdr?/tmp/pip-unpacked-wheel-g426oqom/distributed/protocol/pickle.py_always_use_pickle_fors r)buffer_callbackprotocolc Csg}d|p ti}|ddkr.|dk r.|j|d<zj|tj|f|}d|ksltrtt|ddt krt |dkst |s|t j|f|}Wnbt k rz|t j|f|}Wn2t k r}ztd||W5d}~XYnXYnX|dk r|D]}||q |S) zManage between cloudpickle and pickle 1. Try pickle 2. If it is short then check if it contains __main__ 3. If it is long, then first check type, then check __main__ rNrs__main____name__iz%Failed to serialize %s. Exception: %s)HIGHEST_PROTOCOLappendclearpickledumpsCLOUDPICKLE_GTE_20getattrinspect getmodule cloudpickleZlist_registry_pickle_by_valuelenr Exceptionloggerinfo)rrrbuffersZ dump_kwargsresultebrrrr!s8    rrr)cCsXz$|rtj||dWSt|WSWn.tk rRtjd|ddddYnXdS)Nr-zFailed to deserialize %si'T)exc_info)rloadsr&r'r()rr)rrrr/Dsr/) __future__rr"loggingrr$Zpackaging.versionr parse_version __version__r r getLoggerrr'rrr/rrrrs   #