no-op

PiperOrigin-RevId: 627589952

Author: tfx-copybara

tfx/orchestration/experimental/core/env.py

@@ -14,12 +14,14 @@
 """For environment specific extensions."""
 
 import abc
-from typing import Optional
+from typing import Optional, Sequence
 
 from tfx.orchestration.experimental.core import orchestration_options
 from tfx.proto.orchestration import pipeline_pb2
 from tfx.utils import status as status_lib
 
+from ml_metadata.proto import metadata_store_pb2
+
 _ENV = None
 
 
@@ -85,6 +87,17 @@ def prepare_orchestrator_for_pipeline_run(
       pipeline: The pipeline IR to prepare for.
     """
 
+  @abc.abstractmethod
+  def update_pipeline_run_status(
+      self,
+      owner: str,
+      pipeline_name: str,
+      original_execution: metadata_store_pb2.Execution,
+      modified_execution: metadata_store_pb2.Execution,
+      sub_pipeline_ids: Optional[Sequence[str]] = None,
+  ) -> None:
+    """Updates orchestrator storage backends with pipeline run status."""
+
 
 class _DefaultEnv(Env):
   """Default environment."""
@@ -125,6 +138,16 @@ def prepare_orchestrator_for_pipeline_run(
   ):
     pass
 
+  def update_pipeline_run_status(
+      self,
+      owner: str,
+      pipeline_name: str,
+      original_execution: metadata_store_pb2.Execution,
+      modified_execution: metadata_store_pb2.Execution,
+      sub_pipeline_ids: Optional[Sequence[str]] = None,
+  ) -> None:
+    pass
+
 
 _ENV = _DefaultEnv()
 

tfx/orchestration/experimental/core/env_test.py

@@ -13,12 +13,16 @@
 # limitations under the License.
 """Tests for tfx.orchestration.experimental.core.env."""
 
+from typing import Optional, Sequence
+
 import tensorflow as tf
 from tfx.orchestration.experimental.core import env
 from tfx.orchestration.experimental.core import test_utils
 from tfx.proto.orchestration import pipeline_pb2
 from tfx.utils import status as status_lib
 
+from ml_metadata.proto import metadata_store_pb2
+
 
 class _TestEnv(env.Env):
 
@@ -51,6 +55,16 @@ def prepare_orchestrator_for_pipeline_run(
   ):
     raise NotImplementedError()
 
+  def update_pipeline_run_status(
+      self,
+      owner: str,
+      pipeline_name: str,
+      original_execution: metadata_store_pb2.Execution,
+      modified_execution: metadata_store_pb2.Execution,
+      sub_pipeline_ids: Optional[Sequence[str]] = None,
+  ) -> None:
+    raise NotImplementedError()
+
 
 class EnvTest(test_utils.TfxTest):