Skip to content

Commit 3fd0ab3

Browse files
committed
Fix rerank calculator
1 parent e55b9df commit 3fd0ab3

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/rerank/rerank_calculator.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,6 @@ class RerankCalculator : public CalculatorBase {
5656
static const std::string OUTPUT_TAG_NAME;
5757
static constexpr size_t NUMBER_OF_SPECIAL_TOKENS = 4;
5858

59-
mediapipe::Timestamp timestamp{0};
6059
std::chrono::time_point<std::chrono::system_clock> created;
6160

6261
int64_t bos_token{0};
@@ -403,7 +402,7 @@ class RerankCalculator : public CalculatorBase {
403402
if (!status.ok()) {
404403
return status;
405404
}
406-
cc->Outputs().Tag(OUTPUT_TAG_NAME).Add(new std::string(buffer.GetString()), timestamp);
405+
cc->Outputs().Tag(OUTPUT_TAG_NAME).Add(new std::string(buffer.GetString()), cc->InputTimestamp());
407406
return absl::OkStatus();
408407
} catch (ov::AssertFailure& e) {
409408
SPDLOG_LOGGER_ERROR(rerank_calculator_logger, "OpenVINO Assert Failure: {}", e.what());

0 commit comments

Comments
 (0)