Fix unwanted stage.

sonic-net · Dec 10, 2023 · 9f158c3 · 9f158c3
1 parent 0caaeed
commit 9f158c3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dash-pipeline/SAI/sai_api_gen.py b/dash-pipeline/SAI/sai_api_gen.py
@@ -533,6 +533,9 @@ def parse_p4rt(self, p4rt_table, program, all_actions, ignore_tables):
 
         # The first part of full name is the top level control block name, which is removed for showing better comments as stage.
         self.stage, self.name, self.api_name = self.parse_sai_annotated_name(self.name, full_name_part_start = 1)
+        self.stage = self.stage.replace('.', '_')
+        if "stage" not in self.stage:
+            self.stage = None
 
         # If tables are specified as ignored via CLI or annotations, skip them.
         if self.name in ignore_tables: