From dd0c5a3b7af2f423e03c4feb794cc04f08d507d2 Mon Sep 17 00:00:00 2001 From: Venkat Date: Tue, 25 Jun 2024 11:57:29 -0400 Subject: [PATCH] Integration test env changed --- README.md | 8 ++++---- example_dags/anyscale_job.py | 4 ++-- example_dags/anyscale_service.py | 4 ++-- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index af31f63..19bf22b 100644 --- a/README.md +++ b/README.md @@ -93,8 +93,8 @@ submit_anyscale_job = SubmitAnyscaleJob( task_id="submit_anyscale_job", conn_id=ANYSCALE_CONN_ID, name="AstroJob", - image_uri="anyscale/ray:2.23.0-py311", - compute_config="my-compute-config:1", + image_uri="anyscale/image/airflow-integration-testing:1", + compute_config="airflow-integration-testing:1", working_dir=str(FOLDER_PATH), entrypoint="python ray-job.py", requirements=["requests", "pandas", "numpy", "torch"], @@ -147,8 +147,8 @@ deploy_anyscale_service = RolloutAnyscaleService( task_id="rollout_anyscale_service", conn_id=ANYSCALE_CONN_ID, name=SERVICE_NAME, - image_uri="anyscale/ray:2.23.0-py311", - compute_config="my-compute-config:1", + image_uri="anyscale/image/airflow-integration-testing:1", + compute_config="airflow-integration-testing:1", working_dir="https://github.com/anyscale/docs_examples/archive/refs/heads/main.zip", applications=[{"import_path": "sentiment_analysis.app:model"}], requirements=["transformers", "requests", "pandas", "numpy", "torch"], diff --git a/example_dags/anyscale_job.py b/example_dags/anyscale_job.py index 3b709ac..77c3ff5 100644 --- a/example_dags/anyscale_job.py +++ b/example_dags/anyscale_job.py @@ -33,8 +33,8 @@ task_id="submit_anyscale_job", conn_id=ANYSCALE_CONN_ID, name="AstroJob", - image_uri="anyscale/ray:2.23.0-py311", - compute_config="my-compute-config:1", + image_uri="anyscale/image/airflow-integration-testing:1", + compute_config="airflow-integration-testing:1", working_dir=str(FOLDER_PATH), entrypoint="python ray-job.py", requirements=["requests", "pandas", "numpy", "torch"], diff --git a/example_dags/anyscale_service.py b/example_dags/anyscale_service.py index 720c2aa..a3c803e 100644 --- a/example_dags/anyscale_service.py +++ b/example_dags/anyscale_service.py @@ -34,8 +34,8 @@ task_id="rollout_anyscale_service", conn_id=ANYSCALE_CONN_ID, name=SERVICE_NAME, - image_uri="anyscale/ray:2.23.0-py311", - compute_config="my-compute-config:1", + image_uri="anyscale/image/airflow-integration-testing:1", + compute_config="airflow-integration-testing:1", working_dir="https://github.com/anyscale/docs_examples/archive/refs/heads/main.zip", applications=[{"import_path": "sentiment_analysis.app:model"}], requirements=["transformers", "requests", "pandas", "numpy", "torch"],