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_or_above (model_string : Optional [str ]) -> bool :
@@ -143,7 +143,7 @@ def is_gemini_3_or_above(model_string: Optional[str]) -> bool:
143143 if not model_name .startswith ('gemini-' ):
144144 return False
145145
146- version_string = model_name [len ('gemini-' ):].split ('-' , 1 )[0 ]
146+ version_string = model_name [len ('gemini-' ) :].split ('-' , 1 )[0 ]
147147 if not version_string :
148148 return False
149149
@@ -152,4 +152,4 @@ def is_gemini_3_or_above(model_string: Optional[str]) -> bool:
152152 except InvalidVersion :
153153 return False
154154
155- return parsed_version .major >= 3
155+ return bool ( parsed_version .major >= 3 )
You can’t perform that action at this time.
0 commit comments