
    h&                         S SK r S SKrS SKJrJrJr  S SKJrJr  \R                  R                  \(       + SS9\R                  R                  SS/5      S 5       5       rg)	    N)PyTorchWrapper_v2TorchScriptWrapper_v1pytorch_to_torchscript_wrapper)	has_torchtorchzneeds PyTorch)reasonznN,nI,nO)         c                    [        [        R                  R                  X5      5      R	                  5       n[        U5      n[        R                  R                  X5      R                  S5      nUR                  U5      nUR                  U5      n[        R                  R                  Xg5        UR                  5       n[        5       n	U	R                  U5        [        R                  R                  XiR                  U5      5        g )Nf)r   r   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_model2s
             d/home/james-whalen/.local/lib/python3.13/site-packages/thinc/tests/layers/test_torchscriptwrapper.pytest_pytorch_scriptr%      s     ehhoob56AACE1%8L2"))#.AaA##A&H	MM!!!.&&(J)+MZ(	MM!!!%:%:1%=>    )r   pytest	thinc.apir   r   r   thinc.compatr   r   markskipifparametrizer%    r&   r$   <module>r.      s_      
 * 	M/:i[1? 2 ;?r&   