a
    d                     @   s,   d dl mZ d dlmZ G dd deZdS )    )Any)Metricc                   @   s0   e Zd ZdZeeddddZddddZdS )	_ClassificationTaskWrapperz8Base class for wrapper metrics for classification tasks.N)argskwargsreturnc                 O   s   t | jj ddS )zUpdate metric state.zM metric does not have a global `update` method. Use the task specific metric.NNotImplementedError	__class____name__)selfr   r    r   i/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchmetrics/classification/base.pyupdate   s    z!_ClassificationTaskWrapper.update)r   c                 C   s   t | jj ddS )zCompute metric.zN metric does not have a global `compute` method. Use the task specific metric.Nr   )r   r   r   r   compute   s    z"_ClassificationTaskWrapper.compute)r   
__module____qualname____doc__r   r   r   r   r   r   r   r      s   r   N)typingr   Ztorchmetrics.metricr   r   r   r   r   r   <module>   s   