fix(interactive): Fix metadata getting for loading config (#3915)

As titled.
alibaba · Jul 4, 2024 · fad59b6 · fad59b6
1 parent a99b5bb
commit fad59b6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/flex/storages/rt_mutable_graph/loading_config.cc b/flex/storages/rt_mutable_graph/loading_config.cc
@@ -835,6 +835,9 @@ int32_t LoadingConfig::GetBatchSize() const {
 }
 
 bool LoadingConfig::GetIsBatchReader() const {
+  if (metadata_.find(reader_options::BATCH_READER) == metadata_.end()) {
+    return reader_options::DEFAULT_BATCH_READER;
+  }
   auto str = metadata_.at(reader_options::BATCH_READER);
   return str == "true" || str == "True" || str == "TRUE";
 }

diff --git a/flex/storages/rt_mutable_graph/loading_config.h b/flex/storages/rt_mutable_graph/loading_config.h
@@ -36,6 +36,8 @@ namespace gs {
 
 namespace reader_options {
 static const int32_t DEFAULT_BLOCK_SIZE = (1 << 20);  // 1MB
+static const bool DEFAULT_BATCH_READER =
+    false;  // By default, we read the whole table at once.
 
 // KEY_WORDS for configurations
 static const char* DELIMITER = "delimiter";