File tree 2 files changed +6
-8
lines changed
2 files changed +6
-8
lines changed Original file line number Diff line number Diff line change @@ -228,7 +228,7 @@ awslocal $ENDPOINT_FLAG s3 rm s3://demo-bucket1.0 --recursive --include "*" && \
228
228
229
229
# Test with Polling, no model configuration file - with strict model config disabled
230
230
echo " === Running autocomplete tests ==="
231
- AUTOCOMPLETE_BACKENDS=" libtorch "
231
+ AUTOCOMPLETE_BACKENDS=" onnx "
232
232
export BACKENDS=${AUTOCOMPLETE_BACKENDS}
233
233
234
234
set +e
@@ -241,9 +241,8 @@ for BACKEND in ${AUTOCOMPLETE_BACKENDS}; do
241
241
# Config files specify things expected by unit test like label_filename
242
242
# and max_batch_size for comparing results, so remove some key fields
243
243
# for autocomplete to fill that won't break the unit test.
244
- sed -i ' /platform:/d' models/${model} /config.pbtxt
245
- sed -i ' /data_type:/d' models/${model} /config.pbtxt
246
- sed -i ' /dims:/d' models/${model} /config.pbtxt
244
+ sed -i ' /^input {/,/^}/d' models/${model} /config.pbtxt
245
+ sed -i ' /^output {/,/^}/d' models/${model} /config.pbtxt
247
246
done
248
247
done
249
248
set -e
Original file line number Diff line number Diff line change @@ -252,16 +252,15 @@ sleep 10
252
252
253
253
# Setup model repository with minimal configs to be autocompleted
254
254
rm -rf models && mkdir -p models
255
- AUTOCOMPLETE_BACKENDS=" libtorch_nobatch "
255
+ AUTOCOMPLETE_BACKENDS=" onnx "
256
256
for FW in ${AUTOCOMPLETE_BACKENDS} ; do
257
257
for model in ${FW} _float32_float32_float32 ${FW} _object_object_object; do
258
258
cp -r /data/inferenceserver/${REPO_VERSION} /qa_model_repository/${model} models/
259
259
# Config files specify things expected by unit test like label_filename
260
260
# and max_batch_size for comparing results, so remove some key fields
261
261
# for autocomplete to fill that won't break the unit test.
262
- sed -i ' /platform:/d' models/${model} /config.pbtxt
263
- sed -i ' /data_type:/d' models/${model} /config.pbtxt
264
- sed -i ' /dims:/d' models/${model} /config.pbtxt
262
+ sed -i ' /^input {/,/^}/d' models/${model} /config.pbtxt
263
+ sed -i ' /^output {/,/^}/d' models/${model} /config.pbtxt
265
264
done
266
265
done
267
266
You can’t perform that action at this time.
0 commit comments