torsh-data 0.1.0

Data loading and preprocessing utilities for ToRSh
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

//! # ChainDatasetIter - Trait Implementations
//!
//! This module contains trait implementations for `ChainDatasetIter`.
//!
//! ## Implemented Traits
//!
//! - `Iterator`
//!
//! 🤖 Generated with [SplitRS](https://github.com/cool-japan/splitrs)

use torsh_core::error::Result;

use super::functions::IterableDataset;
use super::types::ChainDatasetIter;

impl<D: IterableDataset + Clone> Iterator for ChainDatasetIter<D> {
    type Item = Result<D::Item>;
    fn next(&mut self) -> Option<Self::Item> {
        loop {
            if let Some(ref mut iter) = self.current_iter {
                if let Some(item) = iter.next() {
                    return Some(item);
                }
            }
            self.current_index += 1;
            if self.current_index >= self.datasets.len() {
                return None;
            }
            self.current_iter = Some(self.datasets[self.current_index].iter());
        }
    }
}