kermit-bench 0.0.2

Benchmarking utilities and benchmarks
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

use {
    super::{benchmark::Benchmark, downloader::Downloader},
    std::path::PathBuf,
};

pub struct DatasetManager {
    dir: PathBuf,
    datasets: Vec<Box<dyn Benchmark + 'static>>,
}

impl DatasetManager {
    pub fn new<P: Into<PathBuf>>(dataset_dir: P) -> Self {
        let path = dataset_dir.into();
        if !path.exists() {
            std::fs::create_dir_all(&path).expect("Failed to create dataset directory");
        }
        Self {
            dir: path,
            datasets: vec![],
        }
    }

    pub fn init_dataset(
        &mut self, dataset: impl Benchmark + 'static,
    ) -> Result<(), Box<dyn std::error::Error>> {
        let dl_spec = &dataset.metadata().download_spec;
        let source = Downloader::download(dl_spec)?;
        dataset.load(&source, self.dir.as_path())?;
        Downloader::clean(dl_spec);
        self.datasets.push(Box::new(dataset));
        Ok(())
    }
}