feat: change gpu_partition_size type hint to str.

PiperOrigin-RevId: 802612342

Author: rui5i

google/cloud/aiplatform/models.py

@@ -4471,7 +4471,7 @@ def deploy(
         max_replica_count: int = 1,
         accelerator_type: Optional[str] = None,
         accelerator_count: Optional[int] = None,
-        gpu_partition_size: Optional[int] = None,
+        gpu_partition_size: Optional[str] = None,
         tpu_topology: Optional[str] = None,
         service_account: Optional[str] = None,
         explanation_metadata: Optional[aiplatform.explain.ExplanationMetadata] = None,
@@ -5733,7 +5733,7 @@ def deploy(
         max_replica_count: int = 1,
         accelerator_type: Optional[str] = None,
         accelerator_count: Optional[int] = None,
-        gpu_partition_size: Optional[int] = None,
+        gpu_partition_size: Optional[str] = None,
         tpu_topology: Optional[str] = None,
         service_account: Optional[str] = None,
         explanation_metadata: Optional[aiplatform.explain.ExplanationMetadata] = None,
@@ -6032,7 +6032,7 @@ def _deploy(
         max_replica_count: int = 1,
         accelerator_type: Optional[str] = None,
         accelerator_count: Optional[int] = None,
-        gpu_partition_size: Optional[int] = None,
+        gpu_partition_size: Optional[str] = None,
         tpu_topology: Optional[str] = None,
         reservation_affinity_type: Optional[str] = None,
         reservation_affinity_key: Optional[str] = None,