umair894/quickstart-trackio
0
1�
2 �1�hi � �, � d dl mZ d dlZ G d� d� Zy)� )�AnyNc �B � e Zd ZdZdZ d3dededz defd�Zde fd �Z4y) � HistogramaL 5 Histogram data type for Trackio, compatible with wandb.Histogram.6 7 Example:8 ```python9 import trackio10 import numpy as np11 12 # Create histogram from sequence13 data = np.random.randn(1000)14 trackio.log({"distribution": trackio.Histogram(data)})15 16 # Create histogram from numpy histogram17 hist, bins = np.histogram(data, bins=30)18 trackio.log({"distribution": trackio.Histogram(np_histogram=(hist, bins))})19 20 # Specify custom number of bins21 trackio.log({"distribution": trackio.Histogram(data, num_bins=50)})22 ```23 24 Args:25 sequence: Optional sequence of values to create histogram from26 np_histogram: Optional pre-computed numpy histogram (hist, bins) tuple27 num_bins: Number of bins for the histogram (default 64, max 512)28 ztrackio.histogramN�sequence� np_histogram�num_binsc �P � |�
|� t d� �|�
|� t d� �t |d� }|�X|\ | _ | _ t j29 | j � | _ t j30 | j � | _ y t j31 |� j
� }|t j |� }t |� dk( r5t j g � | _ t j g � | _ y t j ||�� \ | _ | _ y )Nz,Must provide either sequence or np_histogramz-Cannot provide both sequence and np_histogrami r )�bins)32�33ValueError�min� histogramr34 �np�asarray�flatten�isfinite�len�array)�selfr r r �datas �</usr/local/lib/python3.12/dist-packages/trackio/histogram.py�__init__zHistogram.__init__# s� � � � � � 4� � K�L� L� � �L�$<� � L�M� M��x��%�� � #�(4� %�D�N�D�I� �Z�Z���� 7�D�N��35�36�4�9�9�-�D�I��:�:�h�'�/�/�1�D��� � �D�)�*�D��4�y�A�~�!#���"� ��� �H�H�R�L�� �,.�L�L��H�,M�)���� � �returnc � � | j | j j � | j j � d�S )z,Convert histogram to dictionary for storage.)�_typer37 �values)�TYPEr38 �tolistr
)r s r �_to_dictzHistogram._to_dict> s5 � � �Y�Y��I�I�$�$�&��n�n�+�+�-�39� 40r )NN�@ ) �__name__�41__module__� __qualname__�__doc__r r �tuple�intr �dictr � r r r r sN � ��4 �D� �%)�� N��N� �d�l�N� � N�642�$� 43r r )�typingr �numpyr r r( r r �<module>r+ s �� � �>44� >45r