fix: Fix L0_nullchar_string test (#8120)

Author: krishung5

qa/L0_nullchar_string/nullchar_string_client.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-# Copyright (c) 2019-2020, 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
@@ -89,9 +89,17 @@
 
     # Send inference request to the inference server. Get results for
     # output tensor.
+    input_name = "INPUT0"
+    output_name = "OUTPUT0"
+
+    # If using libtorch model, set input and output name to "INPUT__0" and "OUTPUT__0"
+    if "libtorch" in FLAGS.model_name:
+        input_name = "INPUT__0"
+        output_name = "OUTPUT__0"
+
     inputs = [
         client_util.InferInput(
-            "INPUT0", input0_data.shape, np_to_triton_dtype(np.object_)
+            input_name, input0_data.shape, np_to_triton_dtype(np.object_)
         )
     ]
     inputs[0].set_data_from_numpy(input0_data)
@@ -100,7 +108,7 @@
 
     # We expect there to be 1 result (with batch-size 1). Compare the input
     # and output tensor calculated by the model. They must be the same.
-    output0_data = results.as_numpy("OUTPUT0")
+    output0_data = results.as_numpy(output_name)
 
     print(input0_data, "?=?", output0_data)
     assert np.equal(input0_data.astype(np.bytes_), output0_data).all()

qa/L0_nullchar_string/test.sh

@@ -42,7 +42,7 @@ export CUDA_VISIBLE_DEVICES=0
 
 CLIENT_LOG="./client.log"
 DATADIR=/data/inferenceserver/${REPO_VERSION}/qa_identity_model_repository
-MODELS="onnx_nobatch_zero_1_object libtorch_nobatch_zero_1_object"
+MODELS="python_string libtorch_nobatch_zero_1_object"
 NULLCHAR_CLIENT_PY=nullchar_string_client.py
 
 SERVER=/opt/tritonserver/bin/tritonserver
@@ -53,9 +53,15 @@ source ../common/util.sh
 rm -f $CLIENT_LOG $SERVER_LOG models
 
 mkdir -p models
-for MODEL in $MODELS; do
-    cp -r $DATADIR/$MODEL models/.
-done
+
+# Copy the python model
+mkdir -p models/python_string/1/
+cp -fr ../python_models/string/model.py models/python_string/1/
+cp ../python_models/string/config.pbtxt models/python_string
+sed -i 's/name: "string"/name: "python_string"/' models/python_string/config.pbtxt
+
+# Copy the libtorch model
+cp -r $DATADIR/libtorch_nobatch_zero_1_object models/.
 
 run_server
 if [ "$SERVER_PID" == "0" ]; then