diff --git a/verified_cogen/experiments/incremental_run.py b/verified_cogen/experiments/incremental_run.py index 5d6fd2c..539c7dd 100644 --- a/verified_cogen/experiments/incremental_run.py +++ b/verified_cogen/experiments/incremental_run.py @@ -63,8 +63,9 @@ def main(): args.temperature, ) runner = ValidatingRunner( - wrapping=InvariantRunner(llm, logger, verifier, log_tries), + wrapping=InvariantRunner(llm, logger, verifier), language=language, + log_tries=log_tries ) display_name = rename_file(file) marker_name = str(file.relative_to(directory)) diff --git a/verified_cogen/runners/validating.py b/verified_cogen/runners/validating.py index fe8d8f4..9d77724 100644 --- a/verified_cogen/runners/validating.py +++ b/verified_cogen/runners/validating.py @@ -1,3 +1,4 @@ +import pathlib from typing import Optional from verified_cogen.runners import Runner from verified_cogen.runners.languages.language import Language @@ -9,8 +10,8 @@ class ValidatingRunner(Runner): language: Language prg: Optional[str] = None - def __init__(self, wrapping: Runner, language: Language): - super().__init__(wrapping.llm, wrapping.logger, wrapping.verifier) + def __init__(self, wrapping: Runner, language: Language, log_tries : Optional[pathlib.Path] = None): + super().__init__(wrapping.llm, wrapping.logger, wrapping.verifier, log_tries) self.wrapped_runner = wrapping self.language = language