Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions Dockerfile.QA
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@ RUN apt-get update && \
python3-wheel \
python3-setuptools \
rapidjson-dev \
python3.12-venv \
software-properties-common && \
rm -rf /var/lib/apt/lists/*

Expand All @@ -75,11 +76,11 @@ RUN apt update -q=2 \

# Add inception_graphdef model to example repo
# FIXME: This should be changed to using the fetch_models.sh script
# in order to ensure the public facing docs are up-to-date.
# in order to ensure the public facing docs are up-to-date.
WORKDIR /workspace/docs/examples/model_repository
RUN mkdir -p model_repository/inception_onnx/1 && \
wget -O /tmp/inception_v3_2016_08_28_frozen.pb.tar.gz \
https://storage.googleapis.com/download.tensorflow.org/models/inception_v3_2016_08_28_frozen.pb.tar.gz
https://storage.googleapis.com/download.tensorflow.org/models/inception_v3_2016_08_28_frozen.pb.tar.gz && \
(cd /tmp && tar xzf inception_v3_2016_08_28_frozen.pb.tar.gz) && \
python3 -m venv tf2onnx && \
source ./tf2onnx/bin/activate && \
Expand Down Expand Up @@ -116,7 +117,7 @@ RUN mkdir -p qa/common && \
cp -r docs/examples/model_repository/simple_identity qa/L0_grpc/models && \
cp -r docs/examples/model_repository/simple_sequence qa/L0_grpc/models && \
cp -r docs/examples/model_repository/simple_string qa/L0_grpc/models && \
cp -r docs/examples/model_repository/inception_onnx qa/L0_grpc/models &&
cp -r docs/examples/model_repository/inception_onnx qa/L0_grpc/models && \
mkdir qa/L0_grpc_state_cleanup/models && \
cp -r /workspace/src/test/models/repeat_int32 qa/L0_grpc_state_cleanup/models/ && \
mkdir qa/L0_http/models && \
Expand All @@ -125,7 +126,7 @@ RUN mkdir -p qa/common && \
cp -r docs/examples/model_repository/simple_identity qa/L0_http/models && \
cp -r docs/examples/model_repository/simple_sequence qa/L0_http/models && \
cp -r docs/examples/model_repository/simple_string qa/L0_http/models && \
cp -r docs/examples/model_repository/inception_onnx qa/L0_grpc/models &&
cp -r docs/examples/model_repository/inception_onnx qa/L0_grpc/models && \
mkdir qa/L0_https/models && \
cp -r docs/examples/model_repository/simple qa/L0_https/models/. && \
mkdir qa/L0_secure_grpc/models && \
Expand Down
4 changes: 2 additions & 2 deletions qa/L0_large_payload/test.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#!/bin/bash
# Copyright (c) 2019-2021, NVIDIA CORPORATION. All rights reserved.
# Copyright (c) 2019-2025, NVIDIA CORPORATION. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
Expand Down Expand Up @@ -71,7 +71,7 @@ cp ../python_models/identity_fp32/model.py all_models/python_$MODEL_SUFFIX/1/mod

# Restart server before every test to make sure server state
# is invariant to previous test
for TARGET in1 onnx libtorch plan python; do
for TARGET in onnx libtorch plan python; do
rm -fr models && mkdir models && \
cp -r all_models/${TARGET}_$MODEL_SUFFIX models/.

Expand Down
9 changes: 3 additions & 6 deletions qa/L0_sequence_stress/sequence_stress.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
#!/usr/bin/env python3

# Copyright 2019-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# Copyright 2019-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
Expand Down Expand Up @@ -87,10 +87,7 @@ def check_sequence_async(
(flag_str, value, expected_result, delay_ms)

"""
if (
or ("custom" in trial)
or ("plan" in trial)
):
if ("custom" in trial) or ("plan" in trial):
tensor_shape = (
1,
1,
Expand Down Expand Up @@ -174,7 +171,7 @@ def check_sequence_async(

def get_datatype(trial):
# Get the datatype to use based on what models are available (see test.sh)
if ("plan" in trial):
if "plan" in trial:
return np.float32
return np.int32

Expand Down
Loading