o .wi@s,ddlmZddlmZGdddeZdS))Any)Metricc@s0eZdZdZdededdfddZd dd ZdS) _ClassificationTaskWrapperz8Base class for wrapper metrics for classification tasks.argskwargsreturnNcOt|jjd)zUpdate metric state.zM metric does not have a global `update` method. Use the task specific metric.NotImplementedError __class____name__)selfrrr]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/torchmetrics/classification/base.pyupdate z!_ClassificationTaskWrapper.updatecCr)zCompute metric.zN metric does not have a global `compute` method. Use the task specific metric.r )r rrrcomputerz"_ClassificationTaskWrapper.compute)rN)r __module__ __qualname____doc__rrrrrrrrsrN)typingrtorchmetrics.metricrrrrrrs