Skip to content

Commit 599079b

Browse files
DeanChensjcopybara-github
authored andcommitted
chore: Rename utils.py to gepa_utils.py to ensure consistent first-party module detection for sample scripts across different CI environments
Co-authored-by: Shangjie Chen <deanchen@google.com> PiperOrigin-RevId: 905278958
1 parent 16952bd commit 599079b

3 files changed

Lines changed: 4 additions & 6 deletions

File tree

contributing/samples/gepa/experiment.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
import gepa
3232
from gepa.core.adapter import EvaluationBatch
3333
from gepa.core.adapter import GEPAAdapter
34+
import gepa_utils
3435
from litellm import provider_list
3536
import rater_lib
3637
from retry import retry
@@ -44,8 +45,6 @@
4445
from tau_bench.types import RunConfig
4546
import tau_bench_agent as tau_bench_agent_lib
4647

47-
import utils
48-
4948

5049
def 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
)

contributing/samples/gepa/run_experiment.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,9 @@
2424
from absl import app
2525
from absl import flags
2626
import experiment
27+
import gepa_utils
2728
from 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
)

0 commit comments

Comments
 (0)