Skip to content

Commit 8f2e718

Browse files
authored
Fix compilation error (#1607)
1 parent e045081 commit 8f2e718

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

EventFiltering/PWGHF/HFFilterHelpers.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ std::array<T, 3> PredictONNX(std::vector<T>& inputFeatures, std::shared_ptr<Ort:
234234
}
235235
try {
236236
auto outputTensor = session->Run(session->GetInputNames(), inputTensor, session->GetOutputNames());
237-
assert(outputTensor.size() == outputNames.size() && outputTensor[1].IsTensor());
237+
assert(outputTensor.size() == session->GetOutputNames().size() && outputTensor[1].IsTensor());
238238
auto typeInfo = outputTensor[1].GetTensorTypeAndShapeInfo();
239239
assert(typeInfo.GetElementCount() == 3); // we need multiclass
240240
scores[0] = outputTensor[1].GetTensorMutableData<T>()[0];

0 commit comments

Comments
 (0)