fix server info check

Signed-off-by: Yashash H L <[email protected]>
numaproj · Dec 20, 2024 · 8caf385 · 8caf385
1 parent f9b1b86
commit 8caf385
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/numaflow/src/shared.rs b/numaflow/src/shared.rs
@@ -78,7 +78,10 @@ pub(crate) struct ServerInfo {
 impl ServerInfo {
     pub fn new(container_type: ContainerType) -> Self {
         let mut metadata: HashMap<String, String> = HashMap::new();
-        if container_type == ContainerType::Map || container_type == ContainerType::BatchMap {
+        if container_type == ContainerType::Map
+            || container_type == ContainerType::BatchMap
+            || container_type == ContainerType::MapStream
+        {
             metadata.insert(
                 MAP_MODE_KEY.to_string(),
                 match container_type {