airflow fix

Author: htahir1

airflow-cloud-composer-etl-feature-train/steps/etl/transform.py

@@ -17,7 +17,8 @@
 
 import os
 from datetime import datetime, timezone
-from typing import Annotated, Optional
+from typing import Optional
+from typing_extensions import Annotated
 
 import pandas as pd
 from materializers import BigQueryDataset, CSVDataset

airflow-cloud-composer-etl-feature-train/steps/feature_engineering/augment.py

@@ -17,7 +17,8 @@
 
 import os
 from datetime import datetime, timezone
-from typing import Annotated, Optional
+from typing import Optional
+from typing_extensions import Annotated
 
 from materializers import BigQueryDataset, CSVDataset
 from zenml import step