From db86cef2d308f58325278061c6aa177a535e7e03 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 1 Jun 2022 23:10:12 +0200 Subject: Replace attr with attrs --- text_recognizer/models/metrics.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'text_recognizer/models/metrics.py') diff --git a/text_recognizer/models/metrics.py b/text_recognizer/models/metrics.py index f83c9e4..e59a830 100644 --- a/text_recognizer/models/metrics.py +++ b/text_recognizer/models/metrics.py @@ -1,20 +1,20 @@ """Character Error Rate (CER).""" from typing import Set -import attr +from attrs import define, field import editdistance import torch from torch import Tensor from torchmetrics import Metric -@attr.s(eq=False) +@define(eq=False) class CharacterErrorRate(Metric): """Character error rate metric, computed using Levenshtein distance.""" - ignore_indices: Set[Tensor] = attr.ib(converter=set) - error: Tensor = attr.ib(init=False) - total: Tensor = attr.ib(init=False) + ignore_indices: Set[Tensor] = field(converter=set) + error: Tensor = field(init=False) + total: Tensor = field(init=False) def __attrs_post_init__(self) -> None: super().__init__() -- cgit v1.2.3-70-g09d2