@@ -190,12 +190,29 @@ public JsonObject getModelState(boolean isDefaultVersion) {
190
190
}
191
191
192
192
public void setModelState (JsonObject modelInfo ) {
193
- minWorkers = modelInfo .get (MIN_WORKERS ).getAsInt ();
194
- maxWorkers = modelInfo .get (MAX_WORKERS ).getAsInt ();
195
- maxBatchDelay = modelInfo .get (MAX_BATCH_DELAY ).getAsInt ();
196
- responseTimeout = modelInfo .get (RESPONSE_TIMEOUT ).getAsInt ();
197
- startupTimeout = modelInfo .get (STARTUP_TIMEOUT ).getAsInt ();
198
- batchSize = modelInfo .get (BATCH_SIZE ).getAsInt ();
193
+ minWorkers = modelInfo .has (MIN_WORKERS ) && !modelInfo .get (MIN_WORKERS ).isJsonNull ()
194
+ ? modelInfo .get (MIN_WORKERS ).getAsInt ()
195
+ : 1 ; // default value for minWorkers
196
+
197
+ maxWorkers = modelInfo .has (MAX_WORKERS ) && !modelInfo .get (MAX_WORKERS ).isJsonNull ()
198
+ ? modelInfo .get (MAX_WORKERS ).getAsInt ()
199
+ : 5 ; // default value for maxWorkers
200
+
201
+ maxBatchDelay = modelInfo .has (MAX_BATCH_DELAY ) && !modelInfo .get (MAX_BATCH_DELAY ).isJsonNull ()
202
+ ? modelInfo .get (MAX_BATCH_DELAY ).getAsInt ()
203
+ : 100 ; // default value for maxBatchDelay
204
+
205
+ responseTimeout = modelInfo .has (RESPONSE_TIMEOUT ) && !modelInfo .get (RESPONSE_TIMEOUT ).isJsonNull ()
206
+ ? modelInfo .get (RESPONSE_TIMEOUT ).getAsInt ()
207
+ : 120 ; // default value for responseTimeout
208
+
209
+ startupTimeout = modelInfo .has (STARTUP_TIMEOUT ) && !modelInfo .get (STARTUP_TIMEOUT ).isJsonNull ()
210
+ ? modelInfo .get (STARTUP_TIMEOUT ).getAsInt ()
211
+ : 120 ; // default value for startupTimeout
212
+
213
+ batchSize = modelInfo .has (BATCH_SIZE ) && !modelInfo .get (BATCH_SIZE ).isJsonNull ()
214
+ ? modelInfo .get (BATCH_SIZE ).getAsInt ()
215
+ : 1 ; // default value for batchSize
199
216
200
217
JsonElement runtime = modelInfo .get (RUNTIME_TYPE );
201
218
String runtime_str = Manifest .RuntimeType .PYTHON .getValue ();
0 commit comments