tyadav
diff --git a/‎make_dataset.py
Lines changed: 2 additions & 0 deletions b/‎make_dataset.py
Lines changed: 2 additions & 0 deletions
diff --git a/‎problems.py
Lines changed: 10 additions & 2 deletions b/‎problems.py
Lines changed: 10 additions & 2 deletions
@@ -1,6 +1,7 @@
 import argparse
 import fnmatch
 import time
+import sys
 
 import problems
 import utils
@@ -38,6 +39,7 @@ def main(args):
     probs_by_template = utils.inv_dict({p: p.__module__.split(".")[-1] for p in all_probs})
 
     used_templates = fnmatch.filter(probs_by_template, args.templates)
+    utils.info(f"Python version {sys.version}")
     utils.info(f"Generating from templates: {used_templates}")
     problem_sets = []
     for name in used_templates:  # order determined by import order in templates/__init__.py
 
@@ -732,14 +732,22 @@ class DebugProblem(Problem):
     The program will exit if it successful.
     """
 
-    def add(self, inp: dict, test=True):
+    # def __init__(self):
+    #     print("ho")
+        # super().__init__()
+
 
+    def add(self, inp: dict, test=True):
         if self.check_seen_input(inp):
             return  # don't add duplicate problems
 
         if test:
+            var_name = self.sat_src_spec[1].args[0]
+            type_assertion_str = gen_type_assertion(var_name, self.types[var_name])
             for s in self.sols:
-                assert self.sat(s(**inp), **inp) is True, f"Puzzle {self.name} didn't return True on `{inp}`"
+                answer = s(**inp)
+                exec(type_assertion_str, {var_name: answer})
+                assert self.sat(answer, **inp) is True, f"Puzzle {self.name} didn't return True on `{inp}`"
 
         self.instances.append(("DEBUG TEST", bool(test and self.sols)))  # for counting purposes