From d76ac20e0e84a2ed098f50d8aed3def686ec9311 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Fri, 29 Dec 2023 12:06:38 +0100
Subject: Fix.

---
 candle-book/src/lib.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'candle-book')

diff --git a/candle-book/src/lib.rs b/candle-book/src/lib.rs
index 576ea7f2..f8ca510d 100644
--- a/candle-book/src/lib.rs
+++ b/candle-book/src/lib.rs
@@ -28,6 +28,7 @@ let weights = candle::safetensors::load(weights_filename, &Device::Cpu).unwrap()
     #[rustfmt::skip]
     #[test]
     fn book_hub_2() {
+        {
 // ANCHOR: book_hub_2
 use candle::Device;
 use hf_hub::api::sync::Api;
@@ -43,11 +44,12 @@ let mmap = unsafe { Mmap::map(&file).unwrap() };
 let weights = candle::safetensors::load_buffer(&mmap[..], &Device::Cpu).unwrap();
 // ANCHOR_END: book_hub_2
         assert_eq!(weights.len(), 206);
-    // }
+    }
 
     // #[rustfmt::skip]
     // #[test]
     // fn book_hub_3() {
+    {
 // ANCHOR: book_hub_3
 use candle::{DType, Device, Tensor};
 use hf_hub::api::sync::Api;
@@ -102,6 +104,7 @@ let tp_tensor = Tensor::from_raw_buffer(&raw, dtype, &tp_shape, &Device::Cpu).un
         assert_eq!(view.shape(), &[768, 768]);
         assert_eq!(tp_tensor.dims(), &[192, 768]);
     }
+}
 
     #[rustfmt::skip]
     #[test]
-- 
cgit v1.2.3