Update run_pipeline.py

abhiverse01 · Jul 29, 2024 · 34bc56d · 34bc56d
1 parent 19043d2
commit 34bc56d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scripts/run_pipeline.py b/scripts/run_pipeline.py
@@ -1,5 +1,7 @@
+"""Script to run the TFX pipeline."""
+
+from tfx.orchestration.local import LocalDagRunner
 from pipeline.pipeline import create_pipeline
 
 if __name__ == '__main__':
-    from tfx.orchestration.local import local_dag_runner
-    local_dag_runner.LocalDagRunner().run(create_pipeline())
+    LocalDagRunner().run(create_pipeline())