bug fix in get_group

97df0684 · Guolin Ke · a495a877 · 97df0684
Commit 97df0684 authored Dec 18, 2016 by Guolin Ke
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/io/dataset.cpp src/io/dataset.cpp +1 -1

No files found.
--- a/src/io/dataset.cpp
+++ b/src/io/dataset.cpp
@@ -123,7 +123,7 @@ bool Dataset::GetIntField(const char* field_name, int64_t* out_len, const int**
  name = Common::Trim(name);
  if (name == std::string("query") || name == std::string("group")) {
    *out_ptr = metadata_.query_boundaries();
-    *out_len = metadata_.num_queries();
+    *out_len = metadata_.num_queries() + 1;
  } else {
    return false;
  }