feat: add discard to DialectGroup (#192)

Author: Roger-luo

src/kirin/ir/group.py

@@ -64,8 +64,12 @@ def identity(code: Method):
     def __iter__(self):
         return iter(self.data)
 
+    def __repr__(self) -> str:
+        names = ", ".join(each.name for each in self.data)
+        return f"DialectGroup([{names}])"
+
     @staticmethod
-    def map_module(dialect):
+    def map_module(dialect: Union["Dialect", ModuleType]) -> "Dialect":
         """map the module to the dialect if it is a module.
         It assumes that the module has a `dialect` attribute
         that is an instance of [`Dialect`][kirin.ir.Dialect].
@@ -99,6 +103,26 @@ def union(self, dialect: Iterable[Union["Dialect", ModuleType]]) -> "DialectGrou
             run_pass=self.run_pass_gen,  # pass the run_pass_gen function
         )
 
+    def discard(self, dialect: Union["Dialect", ModuleType]) -> "DialectGroup":
+        """discard a dialect from the group.
+
+        !!! note
+            This does not raise an error if the dialect is not in the group.
+
+        Args:
+            dialect (Union[Dialect, ModuleType]): the dialect to discard
+
+        Returns:
+            DialectGroup: the new dialect group with the discarded dialect.
+        """
+        dialect_ = self.map_module(dialect)
+        return DialectGroup(
+            dialects=frozenset(
+                each for each in self.data if each.name != dialect_.name
+            ),
+            run_pass=self.run_pass_gen,  # pass the run_pass_gen function
+        )
+
     @property
     def registry(self) -> "Registry":
         """return the registry for the dialect group. This

test/ir/test_group.py

@@ -13,3 +13,35 @@ def test_union():
     target_b = group_a.union(group_c)
     assert target_a.data == group_a.data
     assert target_b.data == group_d.data
+
+    target_a_repr = repr(target_a)
+    assert "DialectGroup(" in target_a_repr
+    assert base.dialect.name in target_a_repr
+    assert cf.dialect.name in target_a_repr
+
+    target_b_repr = repr(target_b)
+    assert "DialectGroup(" in target_b_repr
+    assert base.dialect.name in target_b_repr
+    assert cf.dialect.name in target_b_repr
+    assert func.dialect.name in target_b_repr
+
+
+def test_discard():
+    group_a = DialectGroup([base, cf])
+    group_c = DialectGroup([base, func])
+    group_d = DialectGroup([base, func, cf])
+
+    target_a = group_d.discard(cf)
+    target_b = group_d.discard(func)
+    assert target_a.data == group_c.data
+    assert target_b.data == group_a.data
+
+    target_a_repr = repr(target_a)
+    assert "DialectGroup(" in target_a_repr
+    assert base.dialect.name in target_a_repr
+    assert func.dialect.name in target_a_repr
+
+    target_b_repr = repr(target_b)
+    assert "DialectGroup(" in target_b_repr
+    assert base.dialect.name in target_b_repr
+    assert cf.dialect.name in target_b_repr