Update pipeline.py

abhiverse01 · Jul 29, 2024 · 49334ad · 49334ad
1 parent 500343b
commit 49334ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pipeline/pipeline.py b/pipeline/pipeline.py
@@ -3,7 +3,7 @@
 import os
 from tfx.components import CsvExampleGen, StatisticsGen, SchemaGen, ExampleValidator, Transform, Trainer, Evaluator, Pusher
 from tfx.orchestration import pipeline
-from tfx.orchestration.local import local_dag_runner
+from tfx.orchestration.local import LocalDagRunner
 from tfx.orchestration.metadata import sqlite_metadata_connection_config
 from tfx.proto import example_gen_pb2, trainer_pb2, eval_config_pb2, pusher_pb2
 
@@ -89,4 +89,4 @@ def preprocessing_fn(inputs):
     )
 
 if __name__ == '__main__':
-    local_dag_runner.LocalDagRunner().run(create_pipeline())
+    LocalDagRunner().run(create_pipeline())