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 4343from tau_bench .types import EnvRunResult
4444from tau_bench .types import RunConfig
4545import tau_bench_agent as tau_bench_agent_lib
46-
4746import utils
4847
4948
Original file line number Diff line number Diff line change 2525from absl import flags
2626import experiment
2727from google .genai import types
28-
2928import utils
3029
3130_OUTPUT_DIR = flags .DEFINE_string (
Original file line number Diff line number Diff line change @@ -124,7 +124,7 @@ def is_gemini_2_or_above(model_string: Optional[str]) -> bool:
124124 except InvalidVersion :
125125 return False
126126
127- return parsed_version .major >= 2
127+ return bool ( parsed_version .major >= 2 )
128128
129129
130130def is_gemini_3_1_flash_live (model_string : Optional [str ]) -> bool :
@@ -161,7 +161,7 @@ def is_gemini_3_or_above(model_string: Optional[str]) -> bool:
161161 if not model_name .startswith ('gemini-' ):
162162 return False
163163
164- version_string = model_name [len ('gemini-' ):].split ('-' , 1 )[0 ]
164+ version_string = model_name [len ('gemini-' ) :].split ('-' , 1 )[0 ]
165165 if not version_string :
166166 return False
167167
@@ -170,4 +170,4 @@ def is_gemini_3_or_above(model_string: Optional[str]) -> bool:
170170 except InvalidVersion :
171171 return False
172172
173- return parsed_version .major >= 3
173+ return bool ( parsed_version .major >= 3 )
You can’t perform that action at this time.
0 commit comments