work/smartcore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Use usize time for usize::from_le_bytes buffer

Browse Source
This commit is contained in:
Luis Moreno
2021-03-05 10:25:34 -04:00
parent 4335ee5a56
commit 1395cc6518
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/dataset/mod.rs
+2 -1
View File
@@ -82,8 +82,9 @@ pub(crate) fn deserialize_data(
bytes: &[u8], bytes: &[u8],
) -> Result<(Vec<f32>, Vec<f32>, usize, usize), io::Error> { ) -> Result<(Vec<f32>, Vec<f32>, usize, usize), io::Error> {
// read the same file back into a Vec of bytes // read the same file back into a Vec of bytes
const USIZE_SIZE: usize = std::mem::size_of::<usize>();
let (num_samples, num_features) = { let (num_samples, num_features) = {
let mut buffer = [0u8; if cfg!(target_arch = "wasm32") { 4 } else { 8 }]; let mut buffer = [0u8; USIZE_SIZE];
buffer.copy_from_slice(&bytes[0..8]); buffer.copy_from_slice(&bytes[0..8]);
let num_features = usize::from_le_bytes(buffer); let num_features = usize::from_le_bytes(buffer);
buffer.copy_from_slice(&bytes[8..16]); buffer.copy_from_slice(&bytes[8..16]);