allow empty label in rowblackcontainer

dmlc · Dec 27, 2015 · 90c85d9 · 90c85d9 · tqchen · Dec 30, 2015
1 parent f8027b0
commit 90c85d9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/data/row_block.h b/src/data/row_block.h
@@ -137,7 +137,9 @@ template<typename IndexType>
 inline RowBlock<IndexType>
 RowBlockContainer<IndexType>::GetBlock(void) const {
   // consistency check
-  CHECK_EQ(label.size() + 1, offset.size());
+  if (label.size()) {
+    CHECK_EQ(label.size() + 1, offset.size());
+  }
   CHECK_EQ(offset.back(), index.size());
   CHECK(offset.back() == value.size() || value.size() == 0);
   RowBlock<IndexType> data;