Skip to content

Commit

Permalink
Formatting fix
Browse files Browse the repository at this point in the history
  • Loading branch information
gt22 committed Sep 16, 2024
1 parent 73611ff commit e05dc91
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 4 deletions.
4 changes: 3 additions & 1 deletion verified_cogen/args.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,9 @@ def get_default_parser():
"-s", "--output-style", choices=["stats", "full"], default="full"
)
parser.add_argument("--filter-by-ext", help="filter by extension", default=None)
parser.add_argument("--log-tries", help="Save output of every try to given dir", default=None)
parser.add_argument(
"--log-tries", help="Save output of every try to given dir", default=None
)
return parser


Expand Down
2 changes: 1 addition & 1 deletion verified_cogen/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ def main():
),
logger,
verifier,
log_tries
log_tries,
)
for file in files:
with open(file) as f:
Expand Down
12 changes: 10 additions & 2 deletions verified_cogen/runners/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,13 @@ class Runner:
verifier: Verifier
log_tries: Optional[pathlib.Path]

def __init__(self, llm: LLM, logger: Logger, verifier: Verifier, log_tries: Optional[pathlib.Path] = None):
def __init__(
self,
llm: LLM,
logger: Logger,
verifier: Verifier,
log_tries: Optional[pathlib.Path] = None,
):
self.llm = llm
self.logger = logger
self.verifier = verifier
Expand Down Expand Up @@ -83,7 +89,9 @@ def try_fixing(
) -> Optional[int]:
tries = total_tries
while tries > 0:
verification_result = self.verify_program(name, total_tries - tries + 1, inv_prg)
verification_result = self.verify_program(
name, total_tries - tries + 1, inv_prg
)
if verification_result is None:
self.logger.info("Verification timed out")
tries -= 1
Expand Down

0 comments on commit e05dc91

Please sign in to comment.