o  ¢i&ã@sbddlZddlZddlmZmZmZddlmZmZej j e ddej   ddg¡dd „ƒƒZ dS) éN)ÚPyTorchWrapper_v2ÚTorchScriptWrapper_v1Úpytorch_to_torchscript_wrapper)Ú has_torchÚtorchz needs PyTorch)ÚreasonznN,nI,nO)éééc Cs„ttj ||¡ƒ ¡}t|ƒ}tj ||¡  d¡}|  |¡}|  |¡}tj   ||¡|  ¡}tƒ} |  |¡tj   ||   |¡¡dS)NÚf)rrÚnnÚLinearÚ initializerÚnumpyÚrandomÚrandnÚastypeÚpredictÚtestingÚassert_allcloseÚto_bytesrÚ from_bytes) ÚnNÚnIÚnOÚmodelÚ script_modelÚXÚYÚY_scriptÚ serializedÚ script_model2©r"ú^/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/layers/test_torchscriptwrapper.pyÚtest_pytorch_script s   r$) rÚpytestÚ thinc.apirrrÚ thinc.compatrrÚmarkÚskipifÚ parametrizer$r"r"r"r#Ús