burn-dataset 0.22.0

Library with simple dataset APIs for creating ML data pipelines
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

use std::fmt;

/// Default error type used by [`Dataset`](crate::Dataset) implementations that don't define
/// their own.
///
/// Wraps specific dataset error types (e.g. `SqliteDatasetError`) at call sites that don't need
/// the concrete error type. Use `result.map_err(DatasetError::new)?` to convert and propagate an
/// error; `?` alone does not perform this wrapping.
#[derive(Debug)]
pub struct DatasetError(Box<dyn std::error::Error + Send + Sync + 'static>);

impl DatasetError {
    /// Wraps an arbitrary error as a [`DatasetError`].
    pub fn new<E>(err: E) -> Self
    where
        E: std::error::Error + Send + Sync + 'static,
    {
        Self(Box::new(err))
    }
}

impl fmt::Display for DatasetError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.0)
    }
}

impl std::error::Error for DatasetError {
    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
        Some(self.0.as_ref())
    }
}