Fix WASM.

twitchax · Dec 21, 2023 · 8f5242e · 8f5242e
1 parent 8121669
commit 8f5242e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -1,7 +1,9 @@
 {
     "rust-analyzer.cargo.features": [
-        "ml", 
-        "ml_gpu", 
-        "plot"
+        //"ml", 
+        //"ml_gpu", 
+        //"plot",
+        "ml_infer",
+        "wasm"
     ]
 }
diff --git a/Cargo.toml b/Cargo.toml
@@ -50,7 +50,7 @@ ml_train = ["ml_base", "rand", "rayon", "burn-autodiff", "burn/train", "burn/std
 ml_infer = ["ml_base", "burn", "burn-ndarray"]
 ml_gpu = ["ml_train", "burn-tch", "burn-wgpu"]
 
-wasm = ["rodio/wasm-bindgen", "wasm-bindgen", "wasm-bindgen-futures", "js-sys", "console_error_panic_hook", "wee_alloc", "gloo-timers"]
+wasm = ["rodio/wasm-bindgen", "wasm-bindgen", "wasm-bindgen-futures", "js-sys", "console_error_panic_hook", "wee_alloc", "gloo-timers", "burn/wasm-sync"]
 
 plot = ["plotters"]
 

diff --git a/src/wasm.rs b/src/wasm.rs
@@ -2,8 +2,6 @@
 //!
 //! This module contains the WASM wrappers / bindings for the rest of the library.
 
-use std::panic;
-
 use js_sys::{Array, Object, Reflect};
 use wasm_bindgen::{convert::RefFromWasmAbi, prelude::*};