
    h                     $    S SK r S SKJr  S rS rg)    N)	HashEmbedc                      [        SS5      R                  5       n U R                  S5      S:X  d   eU R                  S5      S:X  d   eU R                  S5      R                  S:X  d   eg )N@     nVnOE)r   r   )r   
initializeget_dim	get_paramshape)models    \/home/james-whalen/.local/lib/python3.13/site-packages/thinc/tests/layers/test_hash_embed.py	test_initr      sb    b$**,E==$&&&=="$$$??3%%333    c                     [        SSSS9R                  5       n [        SSSS9R                  5       n[        R                  " SSS9nU R	                  U5      nUR	                  U5      nUR                  5       UR                  5       :w  d   eg )	Nr   r      )seed   )r   uint64)dtype)r   r
   numpyonespredictsum)model1model2arrvector1vector2s        r   test_seed_changes_bucketr!      sv    r4a(335Fr4a(335F
**T
*CnnS!GnnS!G;;=GKKM)))r   )r   	thinc.apir   r   r!    r   r   <module>r$      s     4*r   