Replace attr with attrs

1 files changed, 8 insertions, 5 deletions

diff --git a/text_recognizer/data/iam_paragraphs.py b/text_recognizer/data/iam_paragraphs.py
index 033b93e..b605bbc 100644
--- a/text_recognizer/data/iam_paragraphs.py
+++ b/text_recognizer/data/iam_paragraphs.py
@@ -3,7 +3,7 @@ import json
 from pathlib import Path
 from typing import Dict, List, Optional, Sequence, Tuple
 
-import attr
+from attrs import define, field
 from loguru import logger as log
 import numpy as np
 from PIL import Image, ImageOps
@@ -33,15 +33,15 @@ MAX_LABEL_LENGTH = 682
 MAX_WORD_PIECE_LENGTH = 451
 
 
-@attr.s(auto_attribs=True, repr=False)
+@define(auto_attribs=True, repr=False)
 class IAMParagraphs(BaseDataModule):
     """IAM handwriting database paragraphs."""
 
     # Placeholders
-    dims: Tuple[int, int, int] = attr.ib(
+    dims: Tuple[int, int, int] = field(
         init=False, default=(1, IMAGE_HEIGHT, IMAGE_WIDTH)
     )
-    output_dims: Tuple[int, int] = attr.ib(init=False, default=(MAX_LABEL_LENGTH, 1))
+    output_dims: Tuple[int, int] = field(init=False, default=(MAX_LABEL_LENGTH, 1))
 
     def prepare_data(self) -> None:
         """Create data for training/testing."""
@@ -86,7 +86,10 @@ class IAMParagraphs(BaseDataModule):
                 length=self.output_dims[0],
             )
             return BaseDataset(
-                data, targets, transform=transform, target_transform=target_transform,
+                data,
+                targets,
+                transform=transform,
+                target_transform=target_transform,
             )
 
         log.info(f"Loading IAM paragraph regions and lines for {stage}...")