Skip to content

Commit f4c6c7d

Browse files
committed
precommit mypy fix
1 parent 9a9bb1b commit f4c6c7d

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

codeflash/api/aiservice.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,7 @@ def log_results( # noqa: D417
286286
optimized_runtime: dict[str, float | None] | None,
287287
is_correct: dict[str, bool] | None,
288288
optimized_line_profiler_results: dict[str, str] | None,
289-
metadata: dict[str, any] | None,
289+
metadata: dict[str, Any] | None,
290290
) -> None:
291291
"""Log features to the database.
292292
@@ -299,6 +299,7 @@ def log_results( # noqa: D417
299299
- is_correct (Optional[Dict[str, bool]]): Whether the optimized code is correct.
300300
- optimized_line_profiler_results: line_profiler results for every candidate mapped to their optimization_id
301301
- metadata: contains the best optimization id
302+
302303
"""
303304
payload = {
304305
"trace_id": function_trace_id,

codeflash/optimization/function_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -593,7 +593,7 @@ def determine_best_candidate(
593593
optimized_runtime=optimized_runtimes,
594594
is_correct=is_correct,
595595
optimized_line_profiler_results=optimized_line_profiler_results,
596-
metadata={"best_optimization_id": best_optimization.candidate.optimization_id}
596+
metadata={"best_optimization_id": best_optimization.candidate.optimization_id},
597597
)
598598
return best_optimization
599599

0 commit comments

Comments
 (0)