o
    0ie                     @   sH   d dl Z d dlmZmZ d
ddZdd Zdd Zed	kr"e  dS dS )    N)execute_subprocess_asyncpath_in_accelerate_packagec                 C   sF   | d ur
|  d}ntd}|jdd dd | d ur!|jtd |S )NtestzAccelerate test commandz--config_fileaT  The path to use to store the config file. Will default to a file named default_config.yaml in the cache location, which is the content of the environment `HF_HOME` suffixed with 'accelerate', or if you don't have such an environment variable, your cache directory ('~/.cache' or the content of `XDG_CACHE_HOME`) suffixed with 'huggingface'.)defaulthelp)func)
add_parserargparseArgumentParseradd_argumentset_defaultstest_command)
subparsersparser r   Z/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/accelerate/commands/test.pytest_command_parser   s   
r   c                 C   s`   t ddd}| jd u r|g}nd| j d|  }dg| }t|}|jdkr.td d S d S )	N
test_utilsscriptsztest_script.pyz--config_file= zaccelerate-launchr   z?Test is a success! You are ready for your distributed training!)r   config_filesplitr   
returncodeprint)argsscript_name	test_argscmdresultr   r   r   r   ,   s   


r   c                  C   s   t  } |  }t| d S N)r   
parse_argsr   )r   r   r   r   r   main:   s   r!   __main__r   )r	   accelerate.test_utilsr   r   r   r   r!   __name__r   r   r   r   <module>   s   

