cervo 0.2.0

High level APIs for RL in games.
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::Result;
use clap::Parser;

mod api;
mod describe;
mod package;
mod to_nnef;

/// The command to run.
#[derive(Parser, Debug)]
pub(crate) enum Command {
    ToNnef(to_nnef::ToNnefArgs),
    BatchToNnef(to_nnef::BatchToNnefArgs),
    Api(api::ApiArgs),
    Package(package::PackageArgs),
    Describe(describe::DescribeArgs),
}

pub(crate) fn run(command: Command) -> Result<()> {
    match command {
        Command::ToNnef(config) => to_nnef::onnx_to_nnef(config),
        Command::BatchToNnef(config) => to_nnef::batch_onnx_to_nnef(config),
        Command::Api(config) => api::describe_api(config),
        Command::Describe(config) => describe::describe(config),
        Command::Package(config) => package::package(config),
    }
}