Skip to content

Commit c48ebdf

Browse files
authored
Adjusting storage tests for onnx models (#8128)
1 parent a568240 commit c48ebdf

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

qa/L0_storage_S3_local/test.sh

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -228,7 +228,7 @@ awslocal $ENDPOINT_FLAG s3 rm s3://demo-bucket1.0 --recursive --include "*" && \
228228

229229
# Test with Polling, no model configuration file - with strict model config disabled
230230
echo "=== Running autocomplete tests ==="
231-
AUTOCOMPLETE_BACKENDS="libtorch"
231+
AUTOCOMPLETE_BACKENDS="onnx"
232232
export BACKENDS=${AUTOCOMPLETE_BACKENDS}
233233

234234
set +e
@@ -241,9 +241,8 @@ for BACKEND in ${AUTOCOMPLETE_BACKENDS}; do
241241
# Config files specify things expected by unit test like label_filename
242242
# and max_batch_size for comparing results, so remove some key fields
243243
# 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
247246
done
248247
done
249248
set -e

qa/L0_storage_azure/test.sh

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -252,16 +252,15 @@ sleep 10
252252

253253
# Setup model repository with minimal configs to be autocompleted
254254
rm -rf models && mkdir -p models
255-
AUTOCOMPLETE_BACKENDS="libtorch_nobatch"
255+
AUTOCOMPLETE_BACKENDS="onnx"
256256
for FW in ${AUTOCOMPLETE_BACKENDS}; do
257257
for model in ${FW}_float32_float32_float32 ${FW}_object_object_object; do
258258
cp -r /data/inferenceserver/${REPO_VERSION}/qa_model_repository/${model} models/
259259
# Config files specify things expected by unit test like label_filename
260260
# and max_batch_size for comparing results, so remove some key fields
261261
# 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
265264
done
266265
done
267266

0 commit comments

Comments
 (0)