Skip to content
Open
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion pipeline/workflow/ingestion-helper/clients/spanner.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ class IngestionState(str, Enum):
class IngestionStage(str, Enum):
DATAFLOW = "dataflow"
POSTPROCESSING = "postprocessing"
PREPROCESSING = "preprocessing"


class SpannerClient:
Expand Down Expand Up @@ -323,7 +324,7 @@ def update_ingestion_history(self,
Args:
workflow_id: The ID of the workflow.
status: The status of the ingestion stage (PENDING, RUNNING, SUCCESS, FAILURE).
stage: The stage of the ingestion (dataflow, postprocessing, etc.).
stage: The stage of the ingestion (preprocessing, dataflow, postprocessing, etc.).
job_id: The Dataflow job ID.
ingested_imports: List of ingested import names.
metrics: A dictionary containing metrics about the ingestion.
Expand Down