Fix unwanted stage.

r12f · Dec 9, 2023 · 6f2bd11 · 6f2bd11
1 parent 4d582db
commit 6f2bd11
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
@@ -535,6 +535,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: