cervo 0.1.0

High level APIs for RL in games.
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
36

// Author: Tom Solberg <tom.solberg@embark-studios.com>
// Copyright © 2022, Embark Studios AB, all rights reserved.
// Created: 13 May 2022

/*!

*/

use anyhow::{bail, Result};
use clap::Parser;
use std::{fs::File, path::PathBuf};
use cervo::Inferer;

/// Print API for a model
#[derive(Parser, Debug)]
#[clap()]
pub(crate) struct ImportArgs {
    infile: PathBuf,
    outfile: PathBuf,
}

pub(super) fn describe_api(config: ApiArgs) -> Result<()> {
    let mut reader = File::open(&config.file)?;

    let model = if cervo_nnef::is_nnef_tar(&config.infile) {
		cervo_asset::
    } else {
        match config.file.extension().and_then(|ext| ext.to_str()) {
            Some("onnx") => cervo_onnx::simple_inferer_from_stream(&mut reader)?,
            Some(other) => bail!("unknown file type {:?}", other),
            None => bail!("missing file extension {:?}", config.file),
        }
    };

    Ok(())
}