diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-09 09:03:59 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-09 09:03:59 +0100 |
| commit | 722c50bb0ce18d749edcf86268238e5d9c9ee57e (patch) | |
| tree | 997c32932ff400ac28b6a42e64912a29702deb09 /candle-datasets | |
| parent | 976a1086ee8c30398582df1727ffb27679b2c2d7 (diff) | |
| download | candle-722c50bb0ce18d749edcf86268238e5d9c9ee57e.tar.gz candle-722c50bb0ce18d749edcf86268238e5d9c9ee57e.tar.bz2 candle-722c50bb0ce18d749edcf86268238e5d9c9ee57e.zip | |
Use byteorder in mnist. (#785)
Diffstat (limited to 'candle-datasets')
| -rw-r--r-- | candle-datasets/src/vision/mnist.rs | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/candle-datasets/src/vision/mnist.rs b/candle-datasets/src/vision/mnist.rs index 7a8240cf..2dac883c 100644 --- a/candle-datasets/src/vision/mnist.rs +++ b/candle-datasets/src/vision/mnist.rs @@ -8,10 +8,9 @@ use parquet::file::reader::{FileReader, SerializedFileReader}; use std::fs::File; use std::io::{self, BufReader, Read}; -fn read_u32<T: Read>(reader: &mut T) -> Result<u32> { - let mut b = [0u8; 4]; - reader.read_exact(&mut b)?; - Ok(u32::from_be_bytes(b)) +fn read_u32<T: Read>(reader: &mut T) -> std::io::Result<u32> { + use byteorder::ReadBytesExt; + reader.read_u32::<byteorder::BigEndian>() } fn check_magic_number<T: Read>(reader: &mut T, expected: u32) -> Result<()> { |