@@ -2194,6 +2194,9 @@ public void createBatchPredictionJobTest() throws Exception {
21942194 .setModel (ModelName .of ("[PROJECT]" , "[LOCATION]" , "[MODEL]" ).toString ())
21952195 .setModelVersionId ("modelVersionId-2006125846" )
21962196 .setUnmanagedContainerModel (UnmanagedContainerModel .newBuilder ().build ())
2197+ .setEndpoint (
2198+ EndpointName .ofProjectLocationEndpointName ("[PROJECT]" , "[LOCATION]" , "[ENDPOINT]" )
2199+ .toString ())
21972200 .setInputConfig (BatchPredictionJob .InputConfig .newBuilder ().build ())
21982201 .setInstanceConfig (BatchPredictionJob .InstanceConfig .newBuilder ().build ())
21992202 .setModelParameters (Value .newBuilder ().setBoolValue (true ).build ())
@@ -2269,6 +2272,9 @@ public void createBatchPredictionJobTest2() throws Exception {
22692272 .setModel (ModelName .of ("[PROJECT]" , "[LOCATION]" , "[MODEL]" ).toString ())
22702273 .setModelVersionId ("modelVersionId-2006125846" )
22712274 .setUnmanagedContainerModel (UnmanagedContainerModel .newBuilder ().build ())
2275+ .setEndpoint (
2276+ EndpointName .ofProjectLocationEndpointName ("[PROJECT]" , "[LOCATION]" , "[ENDPOINT]" )
2277+ .toString ())
22722278 .setInputConfig (BatchPredictionJob .InputConfig .newBuilder ().build ())
22732279 .setInstanceConfig (BatchPredictionJob .InstanceConfig .newBuilder ().build ())
22742280 .setModelParameters (Value .newBuilder ().setBoolValue (true ).build ())
@@ -2344,6 +2350,9 @@ public void getBatchPredictionJobTest() throws Exception {
23442350 .setModel (ModelName .of ("[PROJECT]" , "[LOCATION]" , "[MODEL]" ).toString ())
23452351 .setModelVersionId ("modelVersionId-2006125846" )
23462352 .setUnmanagedContainerModel (UnmanagedContainerModel .newBuilder ().build ())
2353+ .setEndpoint (
2354+ EndpointName .ofProjectLocationEndpointName ("[PROJECT]" , "[LOCATION]" , "[ENDPOINT]" )
2355+ .toString ())
23472356 .setInputConfig (BatchPredictionJob .InputConfig .newBuilder ().build ())
23482357 .setInstanceConfig (BatchPredictionJob .InstanceConfig .newBuilder ().build ())
23492358 .setModelParameters (Value .newBuilder ().setBoolValue (true ).build ())
@@ -2418,6 +2427,9 @@ public void getBatchPredictionJobTest2() throws Exception {
24182427 .setModel (ModelName .of ("[PROJECT]" , "[LOCATION]" , "[MODEL]" ).toString ())
24192428 .setModelVersionId ("modelVersionId-2006125846" )
24202429 .setUnmanagedContainerModel (UnmanagedContainerModel .newBuilder ().build ())
2430+ .setEndpoint (
2431+ EndpointName .ofProjectLocationEndpointName ("[PROJECT]" , "[LOCATION]" , "[ENDPOINT]" )
2432+ .toString ())
24212433 .setInputConfig (BatchPredictionJob .InputConfig .newBuilder ().build ())
24222434 .setInstanceConfig (BatchPredictionJob .InstanceConfig .newBuilder ().build ())
24232435 .setModelParameters (Value .newBuilder ().setBoolValue (true ).build ())
0 commit comments