Broken DAG - ModuleNotFoundError: No module named airflow.contrib.kubernetes

Astronomer
1

Hi all,

I’m currently trying to migrate a DAG from version 1.10.10 to version 2.0.0. I got most of it working but my DAG keeps breaking because of this error:

Broken DAG: [/usr/local/airflow/dags/cells_polygon_stores/cells_polygon_stores.py] Traceback (most recent call last):
  File "/usr/local/airflow/dags/cells_polygon_stores/cells_polygon_stores.py", line 34, in <module>
    from dags.utils.ml_operator_sup import MLOperator
  File "/usr/local/airflow/dags/utils/ml_operator_sup.py", line 5, in <module>
    from airflow.contrib.kubernetes.pod import Resources
ModuleNotFoundError: No module named 'airflow.contrib.kubernetes'

I need it for this import:

from airflow.contrib.kubernetes.pod import Resources

I read somewhere that airflow.contrib is deprecated now? What would be the replacement for that module now?

Thanks!