add guri to importer artifacts

PiperOrigin-RevId: 647454281
tensorflow · Jun 28, 2024 · 6b28b63 · 6b28b63
1 parent 4e71a35
commit 6b28b63
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/tfx/orchestration/experimental/core/task_schedulers/importer_task_scheduler.py b/tfx/orchestration/experimental/core/task_schedulers/importer_task_scheduler.py
@@ -23,6 +23,7 @@
 from tfx.utils import status as status_lib
 
 
+
 class ImporterTaskScheduler(task_scheduler.TaskScheduler[task_lib.ExecNodeTask]
                            ):
   """A task scheduler for Importer system node."""
@@ -49,6 +50,7 @@ def schedule(self) -> task_scheduler.TaskSchedulerResult:
         output_key=output_key,
     )
 
+
     return task_scheduler.TaskSchedulerResult(
         status=status_lib.Status(code=status_lib.Code.OK),
         output=task_scheduler.ImporterNodeOutput(

diff --git a/tfx/orchestration/experimental/core/task_schedulers/importer_task_scheduler_test.py b/tfx/orchestration/experimental/core/task_schedulers/importer_task_scheduler_test.py
@@ -31,6 +31,8 @@
 from tfx.utils import status as status_lib
 
 
+
+
 class ImporterTaskSchedulerTest(test_utils.TfxTest):
 
   def setUp(self):
@@ -80,6 +82,7 @@ def _make_pipeline(self, pipeline_root, pipeline_run_id):
     return pipeline
 
   def test_importer_task_scheduler(self):
+
     with self._mlmd_connection as m:
       ts_result = importer_task_scheduler.ImporterTaskScheduler(
           mlmd_handle=m, pipeline=self._pipeline,