Skip to content

Commit 9ad8d3e

Browse files
choich08365chchoi
andauthored
General types of inputs and outputs for MlResponse and OnnxModel (#9319)
Co-authored-by: chchoi <changhwan.choi@cern.ch>
1 parent 660255b commit 9ad8d3e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Tools/ML/MlResponse.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ class MlResponse
158158
LOG(fatal) << "Model index " << nModel << " is out of range! The number of initialised models is " << mModels.size() << ". Please check your configurables.";
159159
}
160160

161-
TypeOutputScore* outputPtr = mModels[nModel].evalModel(input);
161+
TypeOutputScore* outputPtr = mModels[nModel].template evalModel<TypeOutputScore>(input);
162162
return std::vector<TypeOutputScore>{outputPtr, outputPtr + mNClasses};
163163
}
164164

0 commit comments

Comments
 (0)