From a30802ec438cc9da1b439e9897f0de4fa884a5ec Mon Sep 17 00:00:00 2001
From: Luis Moreno <morenoluisma@gmail.com>
Date: Tue, 16 Feb 2021 22:20:02 -0400
Subject: [PATCH] fix: Change to compile for wasm32-unknown-unknown target
 (#80)

---
 src/dataset/mod.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dataset/mod.rs b/src/dataset/mod.rs
index da790b4..31a12cf 100644
--- a/src/dataset/mod.rs
+++ b/src/dataset/mod.rs
@@ -83,7 +83,7 @@ pub(crate) fn deserialize_data(
 ) -> Result<(Vec<f32>, Vec<f32>, usize, usize), io::Error> {
     // read the same file back into a Vec of bytes
     let (num_samples, num_features) = {
-        let mut buffer = [0u8; 8];
+        let mut buffer = [0u8; if cfg!(target_arch = "wasm32") { 4 } else { 8 }];
         buffer.copy_from_slice(&bytes[0..8]);
         let num_features = usize::from_le_bytes(buffer);
         buffer.copy_from_slice(&bytes[8..16]);