o ig@s ddlZddlmZddZdS)N) ParamServercCsLtjddd}||d}||d}t||}|jdksJ|jdks$JdS)N)f)dtype)a)b)r)c)r)numpyzerosr param_keys grad_keys)arrayparamsgradspsrQ/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/backends/test_mem.pytest_param_server_inits    r)r thinc.backends._param_serverrrrrrrs