Shivam249/function-gemma
0
1�
2 jUIi7 � �, � d dl mZ d dlZ G d� d� Zy)� )�SequenceNc �t � e Zd ZdZdZ d3dej ee z ee z dz de4dz de fd�Z de fd �Z
y) � Histograma� 5 Histogram data type for Trackio, compatible with wandb.Histogram.6 7 Args:8 sequence (`np.ndarray` or `Sequence[float]` or `Sequence[int]`, *optional*):9 Sequence of values to create the histogram from.10 np_histogram (`tuple`, *optional*):11 Pre-computed NumPy histogram as a `(hist, bins)` tuple.12 num_bins (`int`, *optional*, defaults to `64`):13 Number of bins for the histogram (maximum `512`).14 15 Example:16 ```python17 import trackio18 import numpy as np19 20 # Create histogram from sequence21 data = np.random.randn(1000)22 trackio.log({"distribution": trackio.Histogram(data)})23 24 # Create histogram from numpy histogram25 hist, bins = np.histogram(data, bins=30)26 trackio.log({"distribution": trackio.Histogram(np_histogram=(hist, bins))})27 28 # Specify custom number of bins29 trackio.log({"distribution": trackio.Histogram(data, num_bins=50)})30 ```31 ztrackio.histogramN�sequence� np_histogram�num_binsc �P � |�
|� t d� �|�
|� t d� �t |d� }|�X|\ | _ | _ t j32 | j � | _ t j33 | j � | _ y t j34 |� 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)35�36ValueError�min� histogramr37 �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��38�39�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.)�_typer40 �values)�TYPEr41 �tolistr
)r s r �_to_dictzHistogram._to_dictA s5 � � �Y�Y��I�I�$�$�&��n�n�+�+�-�42� 43r )NN�@ )�__name__�44__module__� __qualname__�__doc__r r �ndarrayr �float�int�tupler �dictr � r r r r sk � ��: �D� IM�%)�� N��*�*�x���.��#� �>��E�N� �d�l�N� � N�645�$� 46r r )�typingr �numpyr r r* r r �<module>r- s �� � �A47� A48r