add flatmap to set

epfl-lara · Oct 11, 2024 · d7f8c8b · d7f8c8b
1 parent fbdd54b
commit d7f8c8b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/frontends/library/stainless/lang/Set.scala b/frontends/library/stainless/lang/Set.scala
@@ -36,6 +36,11 @@ object Set {
       new Set(set.theSet.map(f))
     }
 
+    @extern @pure
+    def flatMap[B](f: A => Set[B]): Set[B] = {
+      new Set(set.theSet.flatMap(f(_).theSet))
+    }
+
     @extern @pure
     def mapPost1[B](f: A => B)(a: A): Unit = {
       ()