Skip to content

Commit a5f7692

Browse files
committed
fixup! [u r] Add type and domain fields to TDRSourceSpec (#6426)
1 parent b5617d8 commit a5f7692

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

src/azul/terra.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,10 @@
55
from collections.abc import (
66
Sequence,
77
)
8-
import enum
8+
from enum import (
9+
StrEnum,
10+
auto,
11+
)
912
import json
1013
import logging
1114
from time import (
@@ -95,13 +98,13 @@
9598

9699
@attrs.frozen(kw_only=True)
97100
class TDRSourceSpec(SourceSpec):
98-
class Type(enum.Enum):
99-
bigquery = 'bigquery'
100-
parquet = 'parquet'
101+
class Type(StrEnum):
102+
bigquery = auto()
103+
parquet = auto()
101104

102-
class Domain(enum.Enum):
103-
gcp = 'gcp'
104-
azure = 'azure'
105+
class Domain(StrEnum):
106+
gcp = auto()
107+
azure = auto()
105108

106109
type: Type
107110
domain: Domain

0 commit comments

Comments
 (0)