
    hg                         S SK r S SKJr  S rg)    N)ParamServerc                      [         R                  " SSS9n X S.nX S.n[        X5      nUR                  S:X  d   eUR                  S:X  d   eg )N)   f)dtype)a   )b   )r   )c   )r   )numpyzerosr   
param_keys	grad_keys)arrayparamsgradspss       W/home/james-whalen/.local/lib/python3.13/site-packages/thinc/tests/backends/test_mem.pytest_param_server_initr      sQ    KKC(E/F.E	V	#B==0000<<;&&&    )r   thinc.backends._param_serverr   r    r   r   <module>r      s     4'r   