Skip to content

Commit 9857c33

Browse files
committed
setting ai service correctly
1 parent 545b8ed commit 9857c33

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

codeflash/optimization/function_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -364,7 +364,7 @@ def determine_best_candidate(
364364
# Start a new thread for AI service request, start loop in main thread
365365
# check if aiservice request is complete, when it is complete, append result to the candidates list
366366
with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
367-
ai_service_client = self.local_aiservice_client if self.experiment_id else self.aiservice_client
367+
ai_service_client = self.aiservice_client if exp_type=="EXP0" else self.local_aiservice_client
368368
future_line_profile_results = executor.submit(
369369
ai_service_client.optimize_python_code_line_profiler,
370370
source_code=code_context.read_writable_code,

0 commit comments

Comments
 (0)