diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py index 9aabc3ff17..4ac06b8d35 100644 --- a/google/cloud/aiplatform/models.py +++ b/google/cloud/aiplatform/models.py @@ -592,8 +592,8 @@ def _deploy( traffic_percentage: Optional[int] = 0, traffic_split: Optional[Dict[str, int]] = None, machine_type: Optional[str] = None, - min_replica_count: Optional[int] = 1, - max_replica_count: Optional[int] = 1, + min_replica_count: int = 1, + max_replica_count: int = 1, accelerator_type: Optional[str] = None, accelerator_count: Optional[int] = None, service_account: Optional[str] = None, @@ -715,8 +715,8 @@ def _deploy_call( traffic_percentage: Optional[int] = 0, traffic_split: Optional[Dict[str, int]] = None, machine_type: Optional[str] = None, - min_replica_count: Optional[int] = 1, - max_replica_count: Optional[int] = 1, + min_replica_count: int = 1, + max_replica_count: int = 1, accelerator_type: Optional[str] = None, accelerator_count: Optional[int] = None, service_account: Optional[str] = None, @@ -1512,8 +1512,8 @@ def deploy( traffic_percentage: Optional[int] = 0, traffic_split: Optional[Dict[str, int]] = None, machine_type: Optional[str] = None, - min_replica_count: Optional[int] = 1, - max_replica_count: Optional[int] = 1, + min_replica_count: int = 1, + max_replica_count: int = 1, accelerator_type: Optional[str] = None, accelerator_count: Optional[int] = None, service_account: Optional[str] = None, @@ -1649,8 +1649,8 @@ def _deploy( traffic_percentage: Optional[int] = 0, traffic_split: Optional[Dict[str, int]] = None, machine_type: Optional[str] = None, - min_replica_count: Optional[int] = 1, - max_replica_count: Optional[int] = 1, + min_replica_count: int = 1, + max_replica_count: int = 1, accelerator_type: Optional[str] = None, accelerator_count: Optional[int] = None, service_account: Optional[str] = None,