Skip to content

Commit 7653625

Browse files
committed
fix: patch ruff format
1 parent cd359a1 commit 7653625

File tree

3 files changed

+12
-8
lines changed

3 files changed

+12
-8
lines changed

workloads/ingest/ingest/core/config.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,20 +10,20 @@
1010

1111
class DatabaseType(str, Enum):
1212
"""Database type enumeration."""
13-
13+
1414
MYSQL = "mysql"
1515
POSTGRES = "postgres"
1616

1717

1818
class TargetType(str, Enum):
1919
"""Target type enumeration."""
20-
20+
2121
BIGQUERY = "bigquery"
2222

2323

2424
class SourceConfig(BaseModel):
2525
"""Source database configuration."""
26-
26+
2727
name: str
2828
type: DatabaseType
2929
host: str
@@ -51,7 +51,7 @@ def build_connection_string(self, dialect: str, default_port: int = 0) -> str:
5151

5252
class BigQueryTargetConfig(BaseModel):
5353
"""BigQuery target configuration."""
54-
54+
5555
name: str
5656
type: TargetType = TargetType.BIGQUERY
5757
project_id: str = Field(
@@ -77,23 +77,23 @@ class BigQueryTargetConfig(BaseModel):
7777

7878
class RuntimeParams(BaseModel):
7979
"""Runtime parameters for ingestion process."""
80-
80+
8181
retry_attempts: int = Field(ge=1, le=10, default=3)
8282
retry_delay_seconds: int = Field(ge=1, le=3600, default=30)
8383
chunk_size: int = Field(default=10000)
8484

8585

8686
class SecretProvider(str, Enum):
8787
"""Secret provider enumeration."""
88-
88+
8989
GOOGLE_SECRET_MANAGER = "gcloud"
9090
# AWS_SECRET_MANAGER = "aws"
9191
# AZURE_KEY_VAULT = "azure"
9292

9393

9494
class SecretConfig(BaseModel):
9595
"""Secret configuration."""
96-
96+
9797
provider: SecretProvider = Field(
9898
default=SecretProvider.GOOGLE_SECRET_MANAGER, description="Secret Manager Provider"
9999
)
@@ -107,7 +107,7 @@ class SecretConfig(BaseModel):
107107

108108
class Config(BaseModel):
109109
"""Main configuration class."""
110-
110+
111111
version: str
112112
params: RuntimeParams
113113
secrets: List[SecretConfig]

workloads/ingest/ingest/sources/source.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,15 @@ class SourceInterface(Protocol):
2020
def connect(self) -> None:
2121
"""Connect to the data source."""
2222
...
23+
2324
def extract(self, table: Table, chunk_size: int, limit: Optional[int] = None) -> Iterator[dict]:
2425
"""Extract data from the source."""
2526
...
27+
2628
def validate_connection(self) -> bool:
2729
"""Validate the connection to the source."""
2830
...
31+
2932
def close(self) -> None:
3033
"""Close the connection to the source."""
3134
...

workloads/ingest/ingest/targets/target.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ class TargetInterface(Protocol):
1818
def load(self, data: list[dict], target_table: str) -> None:
1919
"""Load data into the target."""
2020
...
21+
2122
def validate_connection(self) -> bool:
2223
"""Validate the connection to the target."""
2324
...

0 commit comments

Comments
 (0)