Check for KVAnnotation instead of deferring to annotationKind().

Signed-off-by: fruffy <[email protected]>
p4lang · Dec 5, 2024 · 4ff9cd5 · 4ff9cd5
1 parent 006177f
commit 4ff9cd5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ir/base.def b/ir/base.def
@@ -355,11 +355,11 @@ class Annotation {
         return expr[idx];
     }
     inline auto &getKV() {
-        BUG_CHECK(annotationKind() == Kind::StructuredKVList, "Annotation does not contain a key-value list.");
+        BUG_CHECK(std::holds_alternative<KVAnnotation>(body), "Annotation does not contain a key-value list.");
         return std::get<KVAnnotation>(body);
     }
     inline const auto &getKV() const {
-        BUG_CHECK(annotationKind() == Kind::StructuredKVList, "Annotation does not contain a key-value list.");
+        BUG_CHECK(std::holds_alternative<KVAnnotation>(body), "Annotation does not contain a key-value list.");
         return std::get<KVAnnotation>(body);
     }