diff --git a/cnpy.h b/cnpy.h index 0d3bb4c..1a82c6d 100644 --- a/cnpy.h +++ b/cnpy.h @@ -85,7 +85,7 @@ namespace cnpy { template<> std::vector& operator+=(std::vector& lhs, const char* rhs); - template void npy_save(std::string fname, const T* data, const std::vector shape, std::string mode = "w") { + template void npy_save(std::string fname, const T* data, const std::vector shape, std::string mode = "w", std::ostream &err = std::clog) { FILE* fp = NULL; std::vector true_data_shape; //if appending, the shape of existing + new data @@ -99,17 +99,17 @@ namespace cnpy { assert(!fortran_order); if(word_size != sizeof(T)) { - std::cout<<"libnpy error: "< void npy_save(std::string fname, const std::vector data, std::string mode = "w") { + template void npy_save(std::string fname, const std::vector data, std::string mode = "w", std::ostream &err = std::clog) { std::vector shape; shape.push_back(data.size()); - npy_save(fname, &data[0], shape, mode); + npy_save(fname, &data[0], shape, mode, err); } template void npz_save(std::string zipname, std::string fname, const std::vector data, std::string mode = "w") {