File tree Expand file tree Collapse file tree
contributing/samples/gepa Expand file tree Collapse file tree Original file line number Diff line number Diff line change 3131import gepa
3232from gepa .core .adapter import EvaluationBatch
3333from gepa .core .adapter import GEPAAdapter
34+ import gepa_utils
3435from litellm import provider_list
3536import rater_lib
3637from retry import retry
4445from tau_bench .types import RunConfig
4546import tau_bench_agent as tau_bench_agent_lib
4647
47- import utils
48-
4948
5049def run_tau_bench_rollouts (
5150 config : RunConfig ,
@@ -583,7 +582,7 @@ def run_gepa(
583582 task_lm = None , # this must be None when a custom adapter is used
584583 adapter = tau_bench_adapter ,
585584 max_metric_calls = config .max_metric_calls ,
586- reflection_lm = utils .reflection_inference_fn (config .reflection_model ),
585+ reflection_lm = gepa_utils .reflection_inference_fn (config .reflection_model ),
587586 reflection_minibatch_size = config .reflection_minibatch_size ,
588587 run_dir = output_dir ,
589588 )
File renamed without changes.
Original file line number Diff line number Diff line change 2424from absl import app
2525from absl import flags
2626import experiment
27+ import gepa_utils
2728from google .genai import types
2829
29- import utils
30-
3130_OUTPUT_DIR = flags .DEFINE_string (
3231 'output_dir' ,
3332 None ,
@@ -106,7 +105,7 @@ def main(argv: Sequence[str]) -> None:
106105 for logger in loggers :
107106 logger .setLevel (logging .WARNING )
108107
109- types .logger .addFilter (utils .FilterInferenceWarnings ())
108+ types .logger .addFilter (gepa_utils .FilterInferenceWarnings ())
110109 output_dir = os .path .join (
111110 _OUTPUT_DIR .value , datetime .now ().strftime ('%Y%m%d%H%M%S%f' )
112111 )
You can’t perform that action at this time.
0 commit comments