summaryrefslogtreecommitdiff
path: root/candle-core/src/pickle.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-core/src/pickle.rs')
-rw-r--r--candle-core/src/pickle.rs2
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/pickle.rs b/candle-core/src/pickle.rs
index 08335257..24f13d20 100644
--- a/candle-core/src/pickle.rs
+++ b/candle-core/src/pickle.rs
@@ -1,4 +1,4 @@
-// Just enough pickle support to be able to read PyTorch checkpoints.
+//! Just enough pickle support to be able to read PyTorch checkpoints.
// This hardcodes objects that are required for tensor reading, we may want to make this a bit more
// composable/tensor agnostic at some point.
use crate::{DType, Error as E, Layout, Result, Tensor};