@@ -393,6 +393,7 @@ def deploy(
393
393
reservation_affinity_key : Optional [str ] = None ,
394
394
reservation_affinity_values : Optional [List [str ]] = None ,
395
395
use_dedicated_endpoint : Optional [bool ] = False ,
396
+ dedicated_endpoint_disabled : Optional [bool ] = False ,
396
397
fast_tryout_enabled : Optional [bool ] = False ,
397
398
system_labels : Optional [Dict [str , str ]] = None ,
398
399
endpoint_display_name : Optional [str ] = None ,
@@ -464,6 +465,10 @@ def deploy(
464
465
use_dedicated_endpoint (bool): Optional. Default value is False. If set
465
466
to True, the underlying prediction call will be made using the
466
467
dedicated endpoint dns.
468
+ dedicated_endpoint_disabled (bool): Optional. Default value is False. If set
469
+ to False, the underlying prediction call will be made using the
470
+ dedicated endpoint dns. Otherwise, the prediction call will be made
471
+ using the shared endpoint dns.
467
472
fast_tryout_enabled (bool): Optional. Defaults to False. If True, model
468
473
will be deployed using faster deployment path. Useful for quick
469
474
experiments. Not for production workloads. Only available for most
@@ -604,9 +609,15 @@ def deploy(
604
609
reservation_affinity_values
605
610
)
606
611
612
+ # TODO(b/417560875): Remove this once notebooks are migrated to use dedicated_endpoint_disabled.
607
613
if use_dedicated_endpoint :
608
614
request .endpoint_config .dedicated_endpoint_enabled = use_dedicated_endpoint
609
615
616
+ if dedicated_endpoint_disabled :
617
+ request .endpoint_config .dedicated_endpoint_disabled = (
618
+ dedicated_endpoint_disabled
619
+ )
620
+
610
621
if fast_tryout_enabled :
611
622
request .deploy_config .fast_tryout_enabled = fast_tryout_enabled
612
623
0 commit comments