Skip to content

Commit 42f0ada

Browse files
send the best optimization id only - not the whole metadata object
1 parent 3562394 commit 42f0ada

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

codeflash/api/aiservice.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@ def log_results( # noqa: D417
305305
original_runtime: float | None,
306306
optimized_runtime: dict[str, float | None] | None,
307307
is_correct: dict[str, bool] | None,
308-
metadata: dict[str, any] | None,
308+
best_optimization_id: str | None,
309309
) -> None:
310310
"""Log features to the database.
311311
@@ -316,7 +316,7 @@ def log_results( # noqa: D417
316316
- original_runtime (Optional[Dict[str, float]]): The original runtime.
317317
- optimized_runtime (Optional[Dict[str, float]]): The optimized runtime.
318318
- is_correct (Optional[Dict[str, bool]]): Whether the optimized code is correct.
319-
- metadata (Optional[dict[str, any]]): metadata.
319+
- best_optimization_id (Optional[str]): The best optimization id.
320320
321321
"""
322322
payload = {
@@ -326,7 +326,7 @@ def log_results( # noqa: D417
326326
"optimized_runtime": optimized_runtime,
327327
"is_correct": is_correct,
328328
"codeflash_version": codeflash_version,
329-
"metadata": metadata,
329+
"best_optimization_id": best_optimization_id,
330330
}
331331
try:
332332
self.make_ai_service_request("/log_features", payload=payload, timeout=5)

codeflash/optimization/function_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -648,7 +648,7 @@ def create_rank_dictionary_compact(int_array: list[int]) -> dict[int, int]:
648648
original_runtime=original_code_baseline.runtime,
649649
optimized_runtime=optimized_runtimes,
650650
is_correct=is_correct,
651-
metadata={"best_optimization_id": best_optimization.candidate.optimization_id},
651+
best_optimization_id=best_optimization.candidate.optimization_id,
652652
)
653653
return best_optimization
654654

0 commit comments

Comments
 (0)