Skip to content

Commit

Permalink
no-op
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 653723298
  • Loading branch information
tfx-copybara committed Jul 31, 2024
1 parent 37e791a commit 272baba
Show file tree
Hide file tree
Showing 2 changed files with 31 additions and 0 deletions.
21 changes: 21 additions & 0 deletions tfx/orchestration/experimental/core/env.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,6 +113,17 @@ def update_pipeline_run_status(
) -> None:
"""Updates orchestrator storage backends with pipeline run status."""

@abc.abstractmethod
def create_pipeline_run_node_executions(
self,
owner: str,
pipeline_name: str,
pipeline: pipeline_pb2.Pipeline,
node_id: str,
executions: Sequence[metadata_store_pb2.Execution],
) -> None:
"""Creates (sub-)pipeline run node executions in the storage backend."""

@abc.abstractmethod
def record_orchestration_time(self, pipeline_run_id: str) -> None:
"""Records the orchestration time for a pipeline run."""
Expand Down Expand Up @@ -211,6 +222,16 @@ def update_pipeline_run_status(
) -> None:
pass

def create_pipeline_run_node_executions(
self,
owner: str,
pipeline_name: str,
pipeline: pipeline_pb2.Pipeline,
node_id: str,
executions: Sequence[metadata_store_pb2.Execution],
) -> None:
pass

def record_orchestration_time(self, pipeline_run_id: str) -> None:
pass

Expand Down
10 changes: 10 additions & 0 deletions tfx/orchestration/experimental/core/env_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,16 @@ def update_pipeline_run_status(
) -> None:
raise NotImplementedError()

def create_pipeline_run_node_executions(
self,
owner: str,
pipeline_name: str,
pipeline: pipeline_pb2.Pipeline,
node_id: str,
executions: Sequence[metadata_store_pb2.Execution],
) -> None:
raise NotImplementedError()

def record_orchestration_time(self, pipeline_run_id: str) -> None:
raise NotImplementedError()

Expand Down

0 comments on commit 272baba

Please sign in to comment.