From 79b0ab13e6d08a12ac0a0971a8001e9ddb8baf56 Mon Sep 17 00:00:00 2001
From: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com>
Date: Wed, 21 Apr 2021 12:31:33 -0400
Subject: [PATCH] fix: remove Optional type hint on deploy (#345)

---
 google/cloud/aiplatform/models.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

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,