diff --git a/RecoTracker/FinalTrackSelectors/plugins/TrackTfClassifier.cc b/RecoTracker/FinalTrackSelectors/plugins/TrackTfClassifier.cc index 75b2e861097eb..6fdec1cd670bc 100644 --- a/RecoTracker/FinalTrackSelectors/plugins/TrackTfClassifier.cc +++ b/RecoTracker/FinalTrackSelectors/plugins/TrackTfClassifier.cc @@ -106,7 +106,7 @@ namespace { std::vector outputs; //evaluate the input - tensorflow::run(const_cast(session_), inputs, {"Identity"}, &outputs); + tensorflow::run(session_, inputs, {"Identity"}, &outputs); for (auto nt = 0; nt < bsize_; nt++) { int itrack = nt + bsize_ * nb; diff --git a/RecoTracker/MkFit/plugins/MkFitOutputConverter.cc b/RecoTracker/MkFit/plugins/MkFitOutputConverter.cc index e7f33abddd957..04263becc38f5 100644 --- a/RecoTracker/MkFit/plugins/MkFitOutputConverter.cc +++ b/RecoTracker/MkFit/plugins/MkFitOutputConverter.cc @@ -715,7 +715,7 @@ std::vector MkFitOutputConverter::computeDNNs(TrackCandidateCollection co //eval and rescale std::vector outputs; - tensorflow::run(const_cast(session), inputs, {"Identity"}, &outputs); + tensorflow::run(session, inputs, {"Identity"}, &outputs); for (auto nt = 0; nt < bsize_; nt++) { int itrack = nt + bsize_ * nb;