Skip to content

Commit

Permalink
Update output_writers.h
Browse files Browse the repository at this point in the history
  • Loading branch information
JesseMckinzie committed Sep 8, 2023
1 parent 3181551 commit 424d926
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/nyx/output_writers.h
Original file line number Diff line number Diff line change
Expand Up @@ -371,16 +371,15 @@ class WriterFactory {
static std::shared_ptr<ApacheArrowWriter> create_writer(const std::string &output_file, const std::vector<std::string> &header) {

if (Nyxus::ends_with_substr(output_file, ".parquet")) {
std::cout << "creating parquet file" << std::endl;

return std::make_shared<ParquetWriter>(output_file);

} else if (Nyxus::ends_with_substr(output_file, ".arrow") || Nyxus::ends_with_substr(output_file, ".feather")) {
std::cout << "creating arrow file" << std::endl;

return std::make_shared<ArrowIPCWriter>(output_file, header);

} else {
std::cout << "error branch" << std::endl;

std::filesystem::path path(output_file);

if (path.has_extension()) {
Expand Down

0 comments on commit 424d926

Please sign in to comment.