diff --git a/src/recordlinker/schemas/identifier.py b/src/recordlinker/schemas/identifier.py index d6ce6c13..cff4e089 100644 --- a/src/recordlinker/schemas/identifier.py +++ b/src/recordlinker/schemas/identifier.py @@ -1,8 +1,10 @@ -import re import enum +import re import typing + import pydantic + class IdentifierType(enum.Enum): """ Enum for various identifier types. @@ -145,6 +147,9 @@ class IdentifierType(enum.Enum): XX = "XX" def __str__(self): + """ + Return the value of the enum as a string. + """ return self.value diff --git a/src/recordlinker/schemas/pii.py b/src/recordlinker/schemas/pii.py index c72b8542..29357ffc 100644 --- a/src/recordlinker/schemas/pii.py +++ b/src/recordlinker/schemas/pii.py @@ -8,7 +8,8 @@ import pydantic from recordlinker import models -from recordlinker.schemas.identifier import Identifier, IdentifierType +from recordlinker.schemas.identifier import Identifier +from recordlinker.schemas.identifier import IdentifierType class FeatureAttribute(enum.Enum):