diff --git a/examples/mnist-inference-web/src/web.rs b/examples/mnist-inference-web/src/web.rs index f5dd233b6..5113f799d 100644 --- a/examples/mnist-inference-web/src/web.rs +++ b/examples/mnist-inference-web/src/web.rs @@ -63,19 +63,6 @@ impl Mnist { mod tests { use super::Mnist; - use burn_dataset::source::huggingface::MNISTDataset; - use burn_dataset::Dataset; - - #[test] - fn inference() { - let dataset = MNISTDataset::test(); - let item = dataset.get(2).unwrap(); - let mnist = Mnist::new(); - let input = item.image.iter().copied().flatten().collect::>(); - let output = mnist.inference(input.as_slice()).unwrap(); - - assert!(output[item.label] > 0.9); - } #[test] fn inference_manual_from_test_data() {