Skip to content

Commit

Permalink
Merge branch 'main' of https://github.com/plasma-umass/coverup into u…
Browse files Browse the repository at this point in the history
…se_model_prefix
  • Loading branch information
emeryberger committed Mar 28, 2024
2 parents 983aa49 + ab8e954 commit 8dccd2a
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions src/coverup/coverup.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,6 @@
from .testrunner import *


PREFIX = 'coverup'

# Turn off most logging
litellm.set_verbose = False
logging.getLogger().setLevel(logging.ERROR)
Expand Down Expand Up @@ -74,7 +72,7 @@ def Path_dir(value):
action=argparse.BooleanOptionalAction,
help=f'show details of lines/branches after each response')

ap.add_argument('--log-file', default=f"{PREFIX}-log",
ap.add_argument('--log-file', default=f"coverup-log",
help='log file to use')

ap.add_argument('--pytest-args', type=str, default='',
Expand All @@ -84,6 +82,9 @@ def Path_dir(value):
action=argparse.BooleanOptionalAction,
help='attempt to install any missing modules')

ap.add_argument('--prefix', type=str, default='coverup',
help='prefix to use for test file names')

ap.add_argument('--write-requirements-to', type=Path,
help='append the name of any missing modules to the given file')

Expand Down Expand Up @@ -112,7 +113,7 @@ def positive_int(value):
def test_file_path(test_seq: int) -> Path:
"""Returns the Path for a test's file, given its sequence number."""
global args
return args.tests_dir / f"test_{PREFIX}_{args.model.split('/')[0]}_{test_seq}.py"
return args.tests_dir / f"test_{args.prefix}_{test_seq}.py"


test_seq = 1
Expand Down

0 comments on commit 8dccd2a

Please sign in to comment.