Fixed arguments

googleapis · Apr 21, 2021 · 30acf42 · 30acf42
1 parent 704ca9f
commit 30acf42
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/samples/model-builder/deploy_model_custom_trained_model_sample.py b/samples/model-builder/deploy_model_custom_trained_model_sample.py
@@ -27,8 +27,8 @@ def deploy_model_custom_trained_model_sample(
     deployed_model_display_name: Optional[str] = None,
     traffic_percentage: Optional[int] = 0,
     traffic_split: Optional[Dict[str, int]] = 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,
     explanation_metadata: Optional[explain.ExplanationMetadata] = None,

diff --git a/samples/model-builder/deploy_model_sample.py b/samples/model-builder/deploy_model_sample.py
@@ -26,10 +26,8 @@ def deploy_model_sample(
     deployed_model_display_name: Optional[str] = None,
     traffic_percentage: Optional[int] = 0,
     traffic_split: Optional[Dict[str, int]] = None,
-    min_replica_count: Optional[int] = 1,
-    max_replica_count: Optional[int] = 1,
-    accelerator_type: Optional[str] = None,
-    accelerator_count: Optional[int] = None,
+    min_replica_count: int = 1,
+    max_replica_count: int = 1,
     explanation_metadata: Optional[explain.ExplanationMetadata] = None,
     explanation_parameters: Optional[explain.ExplanationParameters] = None,
     metadata: Optional[Sequence[Tuple[str, str]]] = (),
@@ -50,8 +48,6 @@ def deploy_model_sample(
         machine_type=None,
         min_replica_count=min_replica_count,
         max_replica_count=max_replica_count,
-        accelerator_type=accelerator_type,
-        accelerator_count=accelerator_count,
         explanation_metadata=explanation_metadata,
         explanation_parameters=explanation_parameters,
         metadata=metadata,