Integration test env changed (#29)

astronomer · Jun 25, 2024 · e820a85 · e820a85
1 parent 9993939
commit e820a85
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git 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
@@ -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
@@ -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"],