diff options
Diffstat (limited to 'utils/Utils.cpp')
-rw-r--r-- | utils/Utils.cpp | 21 |
1 files changed, 2 insertions, 19 deletions
diff --git a/utils/Utils.cpp b/utils/Utils.cpp index b99afb441c..f6aff6f92d 100644 --- a/utils/Utils.cpp +++ b/utils/Utils.cpp @@ -172,30 +172,13 @@ std::tuple<std::vector<unsigned long>, bool, std::string> parse_npy_header(std:: { std::vector<unsigned long> shape; // NOLINT - // Check magic bytes and version number - unsigned char v_major = 0; - unsigned char v_minor = 0; - npy::read_magic(fs, &v_major, &v_minor); - // Read header - std::string header; - if(v_major == 1 && v_minor == 0) - { - header = npy::read_header_1_0(fs); - } - else if(v_major == 2 && v_minor == 0) - { - header = npy::read_header_2_0(fs); - } - else - { - ARM_COMPUTE_ERROR("Unsupported file format version"); - } + std::string header = npy::read_header(fs); // Parse header bool fortran_order = false; std::string typestr; - npy::ParseHeader(header, typestr, &fortran_order, shape); + npy::parse_header(header, typestr, fortran_order, shape); if(!fortran_order) { |