Add get_presemt_trait method to ensure type

Author: david-pl

src/bloqade/squin/analysis/nsites/analysis.py

@@ -23,11 +23,11 @@ def eval_stmt(self, frame: ForwardFrame, stmt: ir.Statement):
         if method is not None:
             return method(self, frame, stmt)
         elif stmt.has_trait(HasSites):
-            has_sites_trait = stmt.get_trait(HasSites)
+            has_sites_trait = stmt.get_present_trait(HasSites)
             sites = has_sites_trait.get_sites(stmt)
             return (NumberSites(sites=sites),)
         elif stmt.has_trait(FixedSites):
-            sites_trait = stmt.get_trait(FixedSites)
+            sites_trait = stmt.get_present_trait(FixedSites)
             return (NumberSites(sites=sites_trait.data),)
         else:
             return (NoSites(),)