
    h                     z   S SK r S SKJrJrJrJr  S SKJrJr  S SK	J
r
  \ R                  S 5       r\ R                  R                  \(       + SS9S 5       r\ R                  R                  \(       + =(       d    \(       + S	S9S
 5       r\ R                  R                  \(       + =(       d    \(       + SS9S 5       rg)    N)has_cupy_gpu	has_torchhas_tensorflowrequire_gpu)use_pytorch_for_gpu_memoryuse_tensorflow_for_gpu_memorycupy_logger_v1c                  .    [        5       n U " S 5      u  pU$ )Nr	   )setup_loggerstep_s      W/home/james-whalen/.local/lib/python3.13/site-packages/spacy_loggers/tests/test_cupy.pyloggerr      s    !#L4 GDK    zCuPy support required)reasonc                 B    [        5         0 nU " U5        US   S:X  d   eg )Ncupy.pool.sourcedefault)r   r   infos     r   "test_cupy_allocator_source_defaultr      s)     MD
4L"#y000r   zCuPy/PyTorch support requiredc                 V    [        5         [        5         0 nU " U5        US   S:X  d   eg )Nr   pytorch)r   r   r   s     r    test_cupy_allocator_source_torchr      s/     M D
4L"#y000r   z CuPy/TensorFlow support requiredc                 V    [        5         [        5         0 nU " U5        US   S:X  d   eg )Nr   
tensorflow)r   r   r   s     r   test_cupy_allocator_source_tfr   $   s/     M!#D
4L"#|333r   )pytest
thinc.utilr   r   r   r   thinc.backendsr   r   spacy_loggers.cupyr
   fixturer   markskipifr   r   r    r   r   <module>r'      s     K K T -   $-DE1 F1 %I.M  11 *N*3U  44r   