torsh-data 0.1.2

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
33
34
35

//! Utility functions and types for collation

use super::{core::DefaultCollate, Collate};
use torsh_core::error::Result;

#[cfg(not(feature = "std"))]
use alloc::boxed::Box;

/// Collation function that can be customized
pub struct CollateFn<F> {
    func: F,
}

impl<F> CollateFn<F> {
    /// Create a new collation function
    pub fn new(func: F) -> Self {
        Self { func }
    }
}

impl<T, O, F> Collate<T> for CollateFn<F>
where
    F: Fn(Vec<T>) -> Result<O>,
{
    type Output = O;

    fn collate(&self, batch: Vec<T>) -> Result<Self::Output> {
        (self.func)(batch)
    }
}

/// Default collation function instance
pub fn collate_fn<T>() -> DefaultCollate {
    DefaultCollate
}