tfx icon indicating copy to clipboard operation
tfx copied to clipboard

tfx.v1 / AirflowDagRunner now inaccessible

Open gbih opened this issue 2 years ago • 0 comments

I am using TFX version: 1.9.1

If I use this import statement to access the Public modules for TFX:

from tfx import v1 as tfx

I noticed that LocalDagRunner and KubeflowDagRunner are defined here https://github.com/tensorflow/tfx/blob/v1.9.1/tfx/v1/orchestration/init.py

and I can still access them, as in

print(tfx.orchestration.LocalDagRunner)
print(tfx.orchestration.experimental.KubeflowDagRunner)
print(tfx.orchestration.experimental.KubeflowV2DagRunner)

However, AirflowDagRunner now seems inaccessible in this v1 namespace.

Is there currently an alternative recommended way to import AirflowDagRunner? All of the previous material I've seen, such as the supporting files for the TFX Airflow Tutorial at https://github.com/tensorflow/tfx/blob/master/tfx/examples/airflow_workshop/setup/dags/taxi_pipeline_solution.py, tend to use the older pattern of

from tfx.orchestration.airflow.airflow_dag_runner import AirflowDagRunner

gbih avatar Aug 15 '22 01:08 gbih