Add initial support for Ada

Add pass@1 metric to pass_k.py

Update pass_k.py to load the results file from .gz or .json

Added basic support for Sets, enabling the translation of
mbpp_473_tuple_intersection.py

Co-authored-by: Rowan Walshe <[email protected]>
Co-authored-by: Fabien Chouteau <[email protected]>

Author: 2 people

dataset_builder/base_language_translator.py

@@ -42,6 +42,13 @@ def gen_dict(self, keys: List[TargetExp], values: List[TargetExp]) -> TargetExp:
         """
         pass
 
+    @abstractmethod
+    def gen_set(self, s: List[TargetExp]) -> TargetExp:
+        """
+        Translate a set with elements s
+        """
+        pass
+
     @abstractmethod
     def gen_call(self, func: TargetExp, args: List[TargetExp]) -> TargetExp:
         """

dataset_builder/generic_translator.py

@@ -27,6 +27,8 @@ def translate_expr(translator, py_expr: ast.AST):
             return translator.gen_list([translate_expr(translator, e) for e in elts])
         case ast.Tuple(elts=elts):
             return translator.gen_tuple([translate_expr(translator, e) for e in elts])
+        case ast.Set(elts=elts):
+            return translator.gen_set([translate_expr(translator, e) for e in elts])
         case ast.Dict(keys=keys, values=values):
             return translator.gen_dict(
                 [translate_expr(translator, e) for e in keys],