Add exception if misesd argument value

triton-inference-server · Jan 7, 2025 · a254b9c · a254b9c
1 parent 5de9816
commit a254b9c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/qa/common/gen_qa_ort_scalar_models.py b/qa/common/gen_qa_ort_scalar_models.py
@@ -124,6 +124,10 @@ def create_onnx_modelconfig(models_dir, dtype, shape):
     )
 
     FLAGS = parser.parse_args()
+
+    if not FLAGS.models_dir:
+        raise Exception("--models_dir is required")
+
     create_onnx_modelfile(FLAGS.models_dir, shape=[1], dtype=np.float32)
     create_onnx_modelconfig(FLAGS.models_dir, shape=[1], dtype=np.float32)
     create_onnx_modelfile(FLAGS.models_dir, shape=[1, 1], dtype=np.float32)