Fix build

Author: meprem

setup.py

@@ -29,7 +29,7 @@ def read(*names, **kwargs):
     "tqdm",
     "pydantic",
     "Jinja2",
-    "pyyaml",
+    "PyYAML==6.0",
     "ploomber>=0.14.6",
     "ploomber-core>=0.0.11",
     # sdist is generated using python -m build, so adding this here.
@@ -60,6 +60,7 @@ def read(*names, **kwargs):
     "kfp",
     # to validate argo specs
     "argo-workflows-dsl",
+    "cython",
     # for testing aws (newer versions break)
     # see: https://github.com/spulec/moto/issues/1793
     "moto==1.3.14",

src/soopervisor/assets/airflow/kubernetes.py

@@ -3,7 +3,7 @@
 
 from airflow import DAG
 from airflow.utils.dates import days_ago
-from airflow.providers.cncf.kubernetes.operators.kubernetes_pod import (
+from airflow.providers.cncf.kubernetes.operators.pod import (
     KubernetesPodOperator,
 )
 

tests/airflow/test_airflow_export.py

@@ -6,7 +6,7 @@
 import json
 
 from airflow import DAG
-from airflow.providers.cncf.kubernetes.operators.kubernetes_pod import (
+from airflow.providers.cncf.kubernetes.operators.pod import (
     KubernetesPodOperator,
 )
 from airflow.operators.bash import BashOperator

tutorials/airflow/ml-intermediate.py

@@ -5,7 +5,7 @@
 
 from airflow import DAG
 from airflow.utils.dates import days_ago
-from airflow.providers.cncf.kubernetes.operators.kubernetes_pod import (
+from airflow.providers.cncf.kubernetes.operators.pod import (
     KubernetesPodOperator,
 )