From 53a4af4ca22ced165fde14cd0de46e29aab7d80d Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 26 Jan 2022 23:17:33 +0100 Subject: fix: change Set to Sequence --- text_recognizer/data/mappings/emnist.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'text_recognizer/data/mappings/emnist.py') diff --git a/text_recognizer/data/mappings/emnist.py b/text_recognizer/data/mappings/emnist.py index 655169e..51e4677 100644 --- a/text_recognizer/data/mappings/emnist.py +++ b/text_recognizer/data/mappings/emnist.py @@ -1,5 +1,5 @@ """Emnist mapping.""" -from typing import List, Optional, Set, Union +from typing import List, Optional, Sequence, Union import torch from torch import Tensor @@ -12,7 +12,7 @@ class EmnistMapping(AbstractMapping): """Mapping for EMNIST labels.""" def __init__( - self, extra_symbols: Optional[Set[str]] = None, lower: bool = True + self, extra_symbols: Optional[Sequence[str]] = None, lower: bool = True ) -> None: self.extra_symbols = set(extra_symbols) if extra_symbols is not None else None self.mapping, self.inverse_mapping, self.input_size = emnist_mapping( -- cgit v1.2.3-70-g09d2