o
    si                     @   sd   d dl Z d dlZd dlZd dlZd dlZd dlZdd Zdd Zdd Z	dd	 Z
d
d Zdd ZdS )    Nc                  C   s*   t   D ]\} }t| d|  qdS )zDCLI function to get info about the Asteroid and dependency versions.20sN)asteroid_versionsitemsprint)kv r   V/home/ubuntu/.local/lib/python3.10/site-packages/asteroid/scripts/asteroid_versions.pyprint_versions	   s   r
   c                   C   s   t  t t dS )N)AsteroidPyTorchzPyTorch-Lightning)asteroid_versionpytorch_versionpytorch_lightning_versionr   r   r   r	   r      s   r   c                   C      t jS N)torch__version__r   r   r   r	   r         r   c                   C   r   r   )plr   r   r   r   r	   r      r   r   c                  C   s8   t tjjj} | d rtj dt|  S tjS )Nz.gitz, Git checkout )	pathlibPath__file__parentjoinpathexistsasteroidr   get_git_version)asteroid_rootr   r   r	   r      s   r   c              
      s    fdd}z|dd}|dddd}|dd}W n t y6 } ztd	| tjd
 W Y d }~dS d }~ww |d d }|rG|d| d7 }|rM|d7 }|S )Nc                     s    t jdg|  d ddS )Ngit)cwdasciiignore)
subprocesscheck_outputstripdecode)cmdrootr   r	   _git(   s    zget_git_version.<locals>._gitz	rev-parseHEADz--symbolic-full-namez--abbrev-refstatusz--porcelainz!Failed to get Git checkout info: )file    z ()z, dirty tree)	Exceptionr   sysstderr)r)   r*   commitbranchdirtyerrsr   r(   r	   r   '   s    
r   )r2   r   r#   r   pytorch_lightningr   r   r
   r   r   r   r   r   r   r   r   r	   <module>   s    