From 33e8565738db91c10b8fda49852edcda7570a73d Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Wed, 8 Mar 2023 13:11:25 +0100
Subject: [PATCH] add validation

---
 launcher/src/main.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/launcher/src/main.rs b/launcher/src/main.rs
index 86c57235..1865cf90 100644
--- a/launcher/src/main.rs
+++ b/launcher/src/main.rs
@@ -148,6 +148,10 @@ fn main() -> ExitCode {
             Some(num_shard) => num_shard,
         }
     };
+    if num_shard < 1 {
+        panic!("`num_shard` cannot be < 1");
+    }
+
     if num_shard > 1 {
         tracing::info!("Sharding model on {num_shard} processes");
     }