From 19e2e2f3fbec32786ee17d238f1a452daca0c67b Mon Sep 17 00:00:00 2001
From: lukebates123 <lucas.bates@hpe.com>
Date: Tue, 9 Jan 2024 13:03:16 -0600
Subject: [PATCH] fix gateway nodes

---
 network_modeling/models/cray-network-architecture.yaml | 2 +-
 network_modeling/models/cray-network-hardware.yaml     | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/network_modeling/models/cray-network-architecture.yaml b/network_modeling/models/cray-network-architecture.yaml
index 53db15527..8b99eb912 100644
--- a/network_modeling/models/cray-network-architecture.yaml
+++ b/network_modeling/models/cray-network-architecture.yaml
@@ -435,7 +435,7 @@ network_v2_tds:
         - "gn"
         - "gateway"
       shasta_name: "gateway"
-      architecture_type: "river_ncn_node_4_port_1g_ocp"
+      architecture_type: "river_ncn_node_4_port"
     - lookup_name:
         - "gpu"
       shasta_name: "gpu"
diff --git a/network_modeling/models/cray-network-hardware.yaml b/network_modeling/models/cray-network-hardware.yaml
index fa0dbb84d..79559e26e 100644
--- a/network_modeling/models/cray-network-hardware.yaml
+++ b/network_modeling/models/cray-network-hardware.yaml
@@ -129,6 +129,9 @@ network_hardware:
       - count: 4
         speed: 1
         slot: ocp
+      - count: 2
+        speed: [25, 1]
+        slot: "pcie-slot1"
       - count: 1
         speed: 1
         slot: "bmc"