Tonic/l-android-control
1
1�
2 �#�h� � �p � d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m3Z4m Z m Z G d� d� � Z
dS )� N)�Client)�
SQLiteStorage)�LogEntry)�
RESERVED_KEYS�fibo�generate_readable_namec �l � e Zd Z ddedededz dedz dedz f5d�Zd� Zd � Zdd6ed e dz fd �Z7d
� Z dS )�RunN�url�project�client�name�configc � � || _ || _ t j � � | _ d | _ || _ |p t t j |� � � � | _8 |pi | _ g | _ t j
� � | _ t j | j �� � | _ d| j _ | j � � � d S )N)�targetT)r r � threading�Lock� _client_lock�_client_thread�_clientr r �get_runsr r � _queued_logs�Event�9_stop_flag�Thread�_init_client_background�daemon�start)�selfr r r
r r s �F/opt/conda/envs/codecanvas/lib/python3.11/site-packages/trackio/run.py�__init__z Run.__init__
s� � � ��� �� � %�N� ,� ,��� "��� �� ��S� 2�=�3I�'�3R�3R� S� S�� � �l��� �,.���#�/�+�+��� '� .�d�6R� S� S� S���%)���"� ��!�!�#�#�#�#�#� c �� � | j � � � s�t j d� � | j 5 | j rr| j �k| j � � � }| j � � � | j � d|t j � � � �� � ddd� � n # 1 swxY w Y | j � � � ��dS dS )z Send batched logs every 500ms.g �?N� /bulk_log��api_name�logs�hf_token)
r �is_set�time�sleepr r r �copy�clear�predict�huggingface_hub�utils� get_token�r � logs_to_sends r �
_batch_senderzRun._batch_sender# s! � ��/�(�(�*�*� � �J�s�O�O�O��"�
�
��$� �� �)A�#'�#4�#9�#9�#;�#;�L��%�+�+�-�-�-��L�(�(�!,� )�!0�!6�!@�!@�!B�!B� )� � � �
�
�
�
�
�
�
�
�
�
�
����
�
�
�
� �/�(�(�*�*� � � � � s �A:B;�;B?�B?c �2 � | j �{t � � }|D ]j} t | j d�� � }| j 5 || _ d d d � � n # 1 swxY w Y n*# t10 $ r Y nw xY w|�t
j d|z � � �k| � � � d S )NF)�verboseg�������?) r r r r r � Exceptionr* r+ r4 )r �fib�sleep_coefficientr
s r r zRun._init_client_background3 s� � � �<� ��&�&�C�%(�
8�
8�!�� #�D�H�e� <� <� <�F� �*� .� .�'-�� �.� .� .� .� .� .� .� .� .� .� .���� .� .� .� .��E�� � � � ��D�����$�0��J�s�%6� 6�7�7�7�� ��� � � � � s4 � A�A � A� A �A�A �A�11A&�%A&�metrics�stepc �* � |� � � D ]2}|t v s|� d� � rt d|� �� � ��3| j | j ||d�}| j 5 | j � |� � d d d � � d S # 1 swxY w Y d S )N�__z1Please do not use this reserved key as a metric: )r �runr: r; ) �keysr �12startswith�13ValueErrorr r r r �append)r r: r; �k� log_entrys r �logzRun.logC s� � �� � � � � � �A��M�!�!�Q�\�\�$�%7�%7�!� �K��K�K�� � � "� �|��9� �� 14� 15� � �
� 0� 0� � � $� $�Y� /� /� /� 0� 0� 0� 0� 0� 0� 0� 0� 0� 0� 0� 0���� 0� 0� 0� 0� 0� 0s � B�B �B c �� � | j � � � | j 5 | j rr| j �k| j � � � }| j � � � | j � d|t j �16 � � �� � ddd� � n # 1 swxY w Y | j �5t d| j
� d�� � | j � d�� � dS dS )z Cleanup when run is finished.Nr$ r% z#* Uploading logs to Trackio Space: z (please wait...)� )�timeout)r �setr r r r, r- r. r/ r0 r1 r �printr �joinr2 s r �finishz17Run.finishT s5 � � ���� � � �
�
� � �� �
�T�\�%=� #� 0� 5� 5� 7� 7� ��!�'�'�)�)�)�� �$�$� (�%� ,� 2� <� <� >� >� %� � � � � � � � � � � � � � ���� � � � � � � *� �S���S�S�S� T� T� T� � � $� $�R� $� 0� 0� 0� 0� 0� +� *s �A:B'�'B+�.B+)NN)N) �__name__�18__module__� __qualname__�strr �dictr! r4 r �intrE rL � r" r r19 r20 s� � � � � � � � "�
$� $�
�$� �$� ��
� $�21 �D�j� $� �t� �
$� $� $� $�,� � � � � � 0� 0�4� 0�s�T�z� 0� 0� 0� 0�"1� 1� 1� 1� 1r" r22 )r r* r/ �
gradio_clientr �trackio.sqlite_storager �trackio.typehintsr �
trackio.utilsr r r r23 rS r" r �<module>rX s� �� � � � � � � � � � � � � � � � � � � 0� 0� 0� 0� 0� 0� &� &� &� &� &� &� E� E� E� E� E� E� E� E� E� E�X1� X1� X1� X1� X1� X1� X1� X1� X1� X1r"