feat: Add ray 2.47 unit tests as required checks

vertex-sdk-bot · copybara-github · commit 5445648eae51 · 2025-07-30T15:47:57.000-07:00
PiperOrigin-RevId: 789072470
diff --git a/.github/sync-repo-settings.yaml b/.github/sync-repo-settings.yaml
@@ -24,6 +24,7 @@ branchProtectionRules:
     - 'Presubmit - Unit Tests Ray 2.33.0'
     - 'Presubmit - Unit Tests Ray 2.42.0 (Python 3.10)'
     - 'Presubmit - Unit Tests Ray 2.42.0 (Python 3.11)'
+    - 'Presubmit - Unit Tests Ray 2.47.1 (Python 3.11)'
     - 'Presubmit - Unit Tests LangChain (Python 3.9)'
     - 'Presubmit - Unit Tests LangChain (Python 3.10)'
     - 'Presubmit - Unit Tests LangChain (Python 3.11)'
diff --git a/.release-please-manifest.json b/.release-please-manifest.json
@@ -1,3 +1,3 @@
 {
-    ".": "1.106.0"
+    ".": "1.105.0"
 }
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,24 +1,5 @@
 # Changelog
 
-## [1.106.0](https://github.com/googleapis/python-aiplatform/compare/v1.105.0...v1.106.0) (2025-07-30)
-
-
-### Features
-
-* Add service_account parameter to AgentEngine class for creation and update ([6359168](https://github.com/googleapis/python-aiplatform/commit/63591683bcf4d3f12fde48b9ea3cc7e075281b0a))
-* Add service_account to Reasoning Engine public protos ([7b6010b](https://github.com/googleapis/python-aiplatform/commit/7b6010b856ac4910cafe49b532ceb34d23fdf63e))
-* Add service_account to Reasoning Engine public protos ([7b6010b](https://github.com/googleapis/python-aiplatform/commit/7b6010b856ac4910cafe49b532ceb34d23fdf63e))
-* Add Vertex AI Model Garden deploy SDK unified Model class in Public Preview ([78c8fdd](https://github.com/googleapis/python-aiplatform/commit/78c8fddc8fc9a9cc83bf982a572ed3aeaf9b876b))
-* Allow adapter_size=32 for supervised tuning ([3a776a7](https://github.com/googleapis/python-aiplatform/commit/3a776a7a41d7b0e8f4a32e5e50401c33bde06985))
-* Vertex AI Model Garden deploy SDK Support for self-deploy Partner Model ([6c72801](https://github.com/googleapis/python-aiplatform/commit/6c728018b09cebe4ca7f7f958df71452acc5e39b))
-
-
-### Bug Fixes
-
-* GenAI SDK client - handle os paths for .whl in requirements when deploying to agent engine ([75cd064](https://github.com/googleapis/python-aiplatform/commit/75cd06442960c5c5a5dd5add8d9e5ae0e9bbfcb9))
-* GenAI SDK client - parsing json body of HTTP response to yield JSON ([8cfd9ba](https://github.com/googleapis/python-aiplatform/commit/8cfd9baef82f83a93b2508584d91c53d11002e88))
-* Use datasets.MultimodalDataset.from_gemini_request_jsonl() without `target_table_id`. ([e92cca2](https://github.com/googleapis/python-aiplatform/commit/e92cca2e85fb5dc796272e70e1c1c8f202c0eafb))
-
 ## [1.105.0](https://github.com/googleapis/python-aiplatform/compare/v1.104.0...v1.105.0) (2025-07-22)
 
 
diff --git a/google/cloud/aiplatform/gapic_version.py b/google/cloud/aiplatform/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/instance/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/instance/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/instance_v1/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/instance_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/params/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/params/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/params_v1/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/params_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/prediction/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/prediction/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/predict/prediction_v1/gapic_version.py b/google/cloud/aiplatform/v1/schema/predict/prediction_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/trainingjob/definition/gapic_version.py b/google/cloud/aiplatform/v1/schema/trainingjob/definition/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/gapic_version.py b/google/cloud/aiplatform/v1/schema/trainingjob/definition_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/instance/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/instance/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/instance_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/params/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/params/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/params_v1beta1/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/params_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/prediction/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/prediction/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/predict/prediction_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/gapic_version.py b/google/cloud/aiplatform/v1beta1/schema/trainingjob/definition_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform/version.py b/google/cloud/aiplatform/version.py
@@ -15,4 +15,4 @@
 # limitations under the License.
 #
 
-__version__ = "1.106.0"
+__version__ = "1.105.0"
diff --git a/google/cloud/aiplatform_v1/gapic_version.py b/google/cloud/aiplatform_v1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/google/cloud/aiplatform_v1beta1/gapic_version.py b/google/cloud/aiplatform_v1beta1/gapic_version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.106.0"  # {x-release-please-version}
+__version__ = "1.105.0"  # {x-release-please-version}
diff --git a/pypi/_vertex_ai_placeholder/version.py b/pypi/_vertex_ai_placeholder/version.py
@@ -15,4 +15,4 @@
 # limitations under the License.
 #
 
-__version__ = "1.106.0"
+__version__ = "1.105.0"
diff --git a/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json b/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-aiplatform",
-    "version": "1.106.0"
+    "version": "0.1.0"
   },
   "snippets": [
     {
diff --git a/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1beta1.json b/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1beta1.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-aiplatform",
-    "version": "1.106.0"
+    "version": "0.1.0"
   },
   "snippets": [
     {
diff --git a/tests/system/vertex_ray/test_cluster_management.py b/tests/system/vertex_ray/test_cluster_management.py
@@ -31,7 +31,7 @@
 class TestClusterManagement(e2e_base.TestEndToEnd):
     _temp_prefix = "temp-rov-cluster-management"
 
-    @pytest.mark.parametrize("cluster_ray_version", ["2.9", "2.33", "2.42"])
+    @pytest.mark.parametrize("cluster_ray_version", ["2.9", "2.33", "2.42", "2.47"])
     def test_cluster_management(self, cluster_ray_version):
         assert ray.__version__ == RAY_VERSION
         aiplatform.init(project=PROJECT_ID, location="us-central1")
diff --git a/tests/system/vertex_ray/test_job_submission_dashboard.py b/tests/system/vertex_ray/test_job_submission_dashboard.py
@@ -35,7 +35,7 @@
 class TestJobSubmissionDashboard(e2e_base.TestEndToEnd):
     _temp_prefix = "temp-job-submission-dashboard"
 
-    @pytest.mark.parametrize("cluster_ray_version", ["2.9", "2.33", "2.42"])
+    @pytest.mark.parametrize("cluster_ray_version", ["2.9", "2.33", "2.42", "2.47"])
     def test_job_submission_dashboard(self, cluster_ray_version):
         assert ray.__version__ == RAY_VERSION
         aiplatform.init(project=PROJECT_ID, location="us-central1")
diff --git a/tests/system/vertex_ray/test_ray_data.py b/tests/system/vertex_ray/test_ray_data.py
@@ -32,37 +32,31 @@
 SDK_VERSION = aiplatform.__version__
 PROJECT_ID = "ucaip-sample-tests"
 
-my_script_ray29 = """
-import ray
-import vertex_ray
 
-parallelism = 10
-query = "SELECT * FROM `bigquery-public-data.ml_datasets.ulb_fraud_detection` LIMIT 10000000"
+def create_bigquery_script(version: str):
+    """Creates a bigquery script for the given Ray version.
 
-ds = vertex_ray.data.read_bigquery(
-    parallelism=parallelism,
-    query=query
-)
+    Args:
+        version: The Ray version.
 
-# The reads are lazy, so the end time cannot be captured until ds.materialize() is called
-ds.materialize()
+    Returns:
+        The bigquery script.
+    """
+    if version == "2.9":
+        num_blocks_arg = "parallelism"
+    else:
+        num_blocks_arg = "override_num_blocks"
+    version_without_dot = version.replace(".", "")
 
-# Write
-vertex_ray.data.write_bigquery(
-    ds,
-    dataset="bugbashbq1.system_test_ray29_write",
-)
-"""
-
-my_script_ray233 = """
+    return f"""
 import ray
 import vertex_ray
 
-override_num_blocks = 10
+{num_blocks_arg} = 10
 query = "SELECT * FROM `bigquery-public-data.ml_datasets.ulb_fraud_detection` LIMIT 10000000"
 
 ds = vertex_ray.data.read_bigquery(
-    override_num_blocks=override_num_blocks,
+    {num_blocks_arg}={num_blocks_arg},
     query=query,
 )
 
@@ -72,39 +66,21 @@
 # Write
 vertex_ray.data.write_bigquery(
     ds,
-    dataset="bugbashbq1.system_test_ray233_write",
+    dataset="bugbashbq1.system_test_ray{version_without_dot}_write",
 )
 """
 
-my_script_ray242 = """
-import ray
-import vertex_ray
-
-override_num_blocks = 10
-query = "SELECT * FROM `bigquery-public-data.ml_datasets.ulb_fraud_detection` LIMIT 10000000"
-
-ds = vertex_ray.data.read_bigquery(
-    override_num_blocks=override_num_blocks,
-    query=query,
-)
-
-# The reads are lazy, so the end time cannot be captured until ds.materialize() is called
-ds.materialize()
-
-# Write
-vertex_ray.data.write_bigquery(
-    ds,
-    dataset="bugbashbq1.system_test_ray242_write",
-)
-"""
 
-my_script = {"2.9": my_script_ray29, "2.33": my_script_ray233, "2.42": my_script_ray242}
+my_script = {
+    version: create_bigquery_script(version)
+    for version in ["2.9", "2.33", "2.42", "2.47"]
+}
 
 
 class TestRayData(e2e_base.TestEndToEnd):
     _temp_prefix = "temp-ray-data"
 
-    @pytest.mark.parametrize("cluster_ray_version", ["2.33", "2.42"])
+    @pytest.mark.parametrize("cluster_ray_version", ["2.33", "2.42", "2.47"])
     def test_ray_data(self, cluster_ray_version):
         head_node_type = vertex_ray.Resources()
         worker_node_types = [

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,3 @@`
`1`	`1`	`{`
`2`		`- ".": "1.106.0"`
	`2`	`+ ".": "1.105.0"`
`3`	`3`	`}`
Original file line number	Diff line number	Diff line change
`@@ -13,4 +13,4 @@`
`13`	`13`	`# See the License for the specific language governing permissions and`
`14`	`14`	`# limitations under the License.`
`15`	`15`	`#`
`16`		`-__version__ = "1.106.0" # {x-release-please-version}`
	`16`	`+__version__ = "1.105.0" # {x-release-please-version}`