john-jac commented on code in PR #66121:
URL: https://github.com/apache/airflow/pull/66121#discussion_r3175548067
##########
providers/amazon/tests/system/amazon/aws/example_s3_vectors.py:
##########
@@ -28,14 +29,27 @@
from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_PLUS
if AIRFLOW_V_3_0_PLUS:
- from airflow.sdk import TriggerRule
+ from airflow.sdk import TriggerRule, task
else:
+ from airflow.decorators import task # type: ignore[attr-defined,no-redef]
from airflow.utils.trigger_rule import TriggerRule # type:
ignore[no-redef,attr-defined]
-DAG_ID = "example_s3vectors"
+DAG_ID = "example_s3_vectors"
sys_test_context_task = SystemTestContextBuilder().build()
+
+@task(trigger_rule=TriggerRule.ALL_DONE)
+def delete_index(vector_bucket_name: str, index_name: str):
+ """Delete the index."""
+ import boto3
+
+ try:
+
boto3.client("s3vectors").delete_index(vectorBucketName=vector_bucket_name,
indexName=index_name)
+ except Exception:
Review Comment:
There was an issue in my ci/cd pipeline that would pull the old version
after ruff tests failed. That's addressed now.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
