Skip to content

Commit

Permalink
run_all_experiments: enable local fuzz introspector instance (#201)
Browse files Browse the repository at this point in the history
Signed-off-by: David Korczynski <[email protected]>
  • Loading branch information
DavidKorczynski authored Apr 3, 2024
1 parent 8ff666e commit 3bbb7ee
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions run_all_experiments.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
from multiprocessing import Pool

import run_one_experiment
from data_prep import introspector
from experiment import benchmark as benchmarklib
from experiment.workdir import WorkDirs
from llm_toolkit import models, prompt_builder
Expand Down Expand Up @@ -157,6 +158,10 @@ def parse_args() -> argparse.Namespace:
action='store_true',
default=False,
help='Add context to function under test.')
parser.add_argument('-e',
'--introspector-endpoint',
type=str,
default=introspector.DEFAULT_INTROSPECTOR_ENDPOINT)
parser.add_argument(
'--delay',
type=int,
Expand Down Expand Up @@ -219,6 +224,8 @@ def main():
experiment_configs = get_experiment_configs(args)
experiment_results = []

introspector.set_introspector_endpoints(args.introspector_endpoint)

print(f'Running {NUM_EXP} experiment(s) in parallel.')
if NUM_EXP == 1:
for config in experiment_configs:
Expand Down

0 comments on commit 3bbb7ee

Please sign in to comment.