From cac51fe16ac1dabb5731b064a20755c8ecc0bc45 Mon Sep 17 00:00:00 2001 From: Congxian Qiu Date: Wed, 28 May 2025 12:13:26 +0800 Subject: [PATCH] (hotfix) fix the doc test for indexer (#2970) --- candle-core/src/indexer.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/candle-core/src/indexer.rs b/candle-core/src/indexer.rs index 2bfaf947..d6cd6deb 100644 --- a/candle-core/src/indexer.rs +++ b/candle-core/src/indexer.rs @@ -226,8 +226,8 @@ where /// assert_eq!(c.to_vec1::()?, &[1., 4.]); /// /// let d = a.i((2.., ..))?; - /// assert_eq!(c.shape().dims(), &[2]); - /// assert_eq!(c.to_vec1::()?, &[1., 4.]); + /// assert_eq!(d.shape().dims(), &[1, 3]); + /// assert_eq!(d.to_vec2::()?, &[[6., 7., 8.]]); /// # Ok::<(), candle_core::Error>(()) /// ``` fn i(&self, (a, b): (A, B)) -> Result {