Skip to content

Commit

Permalink
Update typing in somajo.py to use Literal (#115)
Browse files Browse the repository at this point in the history
  • Loading branch information
PhilipMay authored Dec 21, 2023
1 parent b64d0f9 commit d02e5c4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mltb2/somajo.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@

from abc import ABC
from dataclasses import dataclass, field
from typing import Container, Dict, Iterable, List, Optional, Set, Tuple, Union
from typing import Container, Dict, Iterable, List, Literal, Optional, Set, Tuple, Union

from somajo import SoMaJo
from tqdm import tqdm
Expand All @@ -29,7 +29,7 @@ class SoMaJoBaseClass(ABC):
This class is an abstract base class. It should not be used directly.
"""

language: str
language: Literal["de_CMC", "en_PTB"]
somajo: SoMaJo = field(init=False, repr=False)

def __post_init__(self):
Expand Down

0 comments on commit d02e5c4

Please sign in to comment.