Merge pull request #1986 from iamsusiep:dummy

PiperOrigin-RevId: 319701322
tensorflow · Jul 14, 2020 · 9352486 · 9352486
2 parents b672db7 + ca9c62f
commit 9352486
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tfx/experimental/pipeline_testing/stub_component_launcher.py b/tfx/experimental/pipeline_testing/stub_component_launcher.py
@@ -69,14 +69,16 @@ def get_stub_launcher_class(
   supported by the pipeline.
   For example:
     class MyPusherStubExecutor(base_stub_executor.BaseStubExecutor){...}
+    class MyTransformStubExecutor(base_stub_executor.BaseStubExecutor){...}
 
     PipelineConfig(
         supported_launcher_classes=[
             stub_component_launcher.get_stub_launcher_class(
                 test_data_dir,
                 stubbed_component_ids = ['CsvExampleGen'],
-                stubbed_component_map = {'Pusher':
-                                          MyPusherStubExecutor})
+                stubbed_component_map = {
+                    'Transform': MyTransformStubExecutor,
+                    'Pusher': MyPusherStubExecutor})
         ],
     )
   The method also sets the necessary class variables for the