
    ukir                     H   U d dl mZ d dlZd dlZd dlZdgZ ej                  e      Ze	e
d<    G d de      Zee
d<   	 d dlmZ ej                  d       ej                  Zd	Zdedej                  dz  fdZy# e$ r" ej                  d
       ej                  ZdZY <w xY w)    )ProtocolNPathepath_installedc                   N    e Zd ZdZdeej                  z  dej                  fdZ	y)PathProtocolz"A factory that creates a PurePath.pathsegmentsreturnc                      y )N )selfr   s     H/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/path.py__call__zPathProtocol.__call__   s        N)
__name__
__module____qualname____doc__strosPathLikepathlibr   r   r   r   r   r   r      s%    *C"++$5 ',, r   r   )epathz2etils.epath found. Using etils.epath for file I/O.Tz6etils.epath was not found. Using pathlib for file I/O.Fout_dir_pathr	   c                     | sy| dk(  r-t         j                  j                  dd      } | st        d      t	        |       }|j                  dd       |S )zHMake a directory or return the undeclared outputs directory if `sponge`.NspongeTEST_UNDECLARED_OUTPUTS_DIR zhGot output directory (e.g., via JAX_DUMP_IR_TO) 'sponge' but TEST_UNDECLARED_OUTPUTS_DIR is not defined.T)parentsexist_ok)r   environget
ValueErrorr   mkdir)r   out_dirs     r   make_jax_dump_dirr%   /   s[    	X::>>"?DL9  '	--t-,	.r   )typingr   loggingr   r   __all__	getLoggerr   loggerbool__annotations__r   etilsr   debugr   r   ImportErrorr   r%   r   r   r   <module>r0      s      	 (			8	$ 8 
  
 	,,CD 
$/C GLL4,?   ,,GH	$/s   A: :$B! B!