fixes

JetBrains-Research · Nov 29, 2024 · 60b1475 · 60b1475
1 parent e8430dc
commit 60b1475
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/verified_cogen/main.py b/verified_cogen/main.py
@@ -131,7 +131,10 @@ def make_runner_cls(
     bench_type: str, extension: str, config: RunnerConfig
 ) -> Callable[[LLM, Logger, Verifier, Optional[Rewriter]], Runner]:
     def runner_cls(
-        llm: LLM, logger: Logger, verifier: Verifier, rewriter: Optional[Rewriter]
+        llm: LLM,
+        logger: Logger,
+        verifier: Verifier,
+        rewriter: Optional[Rewriter] = None,
     ):
         if bench_type == "invariants":
             return InvariantRunner(llm, logger, verifier, config)

diff --git a/verified_cogen/runners/validating.py b/verified_cogen/runners/validating.py
@@ -11,7 +11,7 @@ class ValidatingRunner(Runner):
     wrapped_runner: Runner
     language: Language
     summarizer_llm: LLM
-    pure_non_helpers: [str] = []
+    pure_non_helpers: [str]
 
     def __init__(
         self,
@@ -32,6 +32,7 @@ def __init__(
         )
         self.wrapped_runner = wrapping
         self.language = language
+        self.pure_non_helpers = []
 
     def _add_validators(self, prg: str, inv_prg: str):
         validators = self.language.generate_validators(prg)