Fix context graph (#292)

2023-08-28 19:39:22 +08:00
parent 49ec7e8f57
commit 2b0152d2a2
3 changed files with 10 additions and 18 deletions
--- a/sherpa-onnx/csrc/context-graph-test.cc
+++ b/sherpa-onnx/csrc/context-graph-test.cc
@@ -22,8 +22,9 @@ TEST(ContextGraph, TestBasic) {
  auto context_graph = ContextGraph(contexts, 1);

  auto queries = std::map<std::string, float>{
-      {"HEHERSHE", 14}, {"HERSHE", 12}, {"HISHE", 9},   {"SHED", 6},
-      {"HELL", 2},      {"HELLO", 7},   {"DHRHISQ", 4}, {"THEN", 2}};
+      {"HEHERSHE", 14}, {"HERSHE", 12}, {"HISHE", 9},
+      {"SHED", 6},      {"SHELF", 6},   {"HELL", 2},
+      {"HELLO", 7},     {"DHRHISQ", 4}, {"THEN", 2}};

  for (const auto &iter : queries) {
    float total_scores = 0;