generated from OBJNULL/Dockerized-Rust
Fixed type error
This commit is contained in:
parent
8b918113ac
commit
31c30ed200
1 changed files with 1 additions and 1 deletions
|
@ -13,7 +13,7 @@ pub fn get_operation_mode() -> Option<OperationMode> {
|
|||
let args: Vec<String> = env::args().collect();
|
||||
|
||||
// Getting operation mode
|
||||
match &args[1] {
|
||||
match args[1].as_str() {
|
||||
"training" => Some(OperationMode::Training),
|
||||
"infrence" => Some(OperationMode::Infrence),
|
||||
_ => None,
|
||||
|
|
Loading…
Reference in a new issue