Skip to content

Commit 679b2fb

Browse files
committed
Fix testInference() unitest after declaring inputs as mandatory parameter
1 parent 1f8f0fb commit 679b2fb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/src/unittests/machinelearning/test_onnxpredict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ def runEffnetDiscogsInference(onnx_predict, outputs, pool) -> Pool:
227227

228228
onnx_predict.configure(
229229
graphFilename= model,
230-
inputs=[],
230+
inputs=["melspectrogram"],
231231
outputs=[output["name"] for output in outputs],
232232
)
233233

0 commit comments

Comments
 (0)