fix enum warning

This commit is contained in:
Aseem Bansal 2025-06-26 16:34:48 +05:30
parent 96177f9f5f
commit 4fe28bdb07

View File

@ -1,5 +1,4 @@
import logging import logging
from enum import Enum
from typing import Dict, Iterable, List, Optional, Tuple from typing import Dict, Iterable, List, Optional, Tuple
from pydantic import Field from pydantic import Field
@ -24,11 +23,12 @@ from datahub.metadata.schema_classes import (
UpstreamClass, UpstreamClass,
UpstreamLineageClass, UpstreamLineageClass,
) )
from datahub.utilities.str_enum import StrEnum
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class SubTypePattern(str, Enum): class SubTypePattern(StrEnum):
ALTERNATING = "alternating" ALTERNATING = "alternating"
ALL_TABLE = "all_table" ALL_TABLE = "all_table"
ALL_VIEW = "all_view" ALL_VIEW = "all_view"