Skip to content

Commit ee12aee

Browse files
authored
fix: window -> pub window in dataset/mod.rs (#1658)
* update dataset/mod.rs * Update mod.rs * Update window.rs
1 parent 9fbcbed commit ee12aee

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

crates/burn-dataset/src/dataset/mod.rs

+1
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,4 @@ pub use in_memory::*;
1414
pub use iterator::*;
1515
#[cfg(any(feature = "sqlite", feature = "sqlite-bundled"))]
1616
pub use sqlite::*;
17+
pub use window::*;

crates/burn-dataset/src/dataset/window.rs

+2-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ impl<'a, I> WindowDataset<'a, I> {
99
/// # Parameters
1010
///
1111
/// - `dataset`: The dataset over which windows will be created.
12-
/// - `size`: The size of the window.
12+
/// - `size`: The size of the windows.
1313
///
1414
/// # Returns
1515
///
@@ -24,6 +24,7 @@ impl<'a, I> WindowDataset<'a, I> {
2424

2525
/// Dataset designed to work with overlapping windows of data.
2626
pub struct WindowDataset<'a, I> {
27+
/// The size of the windows.
2728
pub size: NonZeroUsize,
2829
dataset: &'a dyn Dataset<I>,
2930
}

0 commit comments

Comments
 (0)