tfx
tfx copied to clipboard
tfx.v1 / AirflowDagRunner now inaccessible
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