Skip to content

Commit ab7ef24

Browse files
committed
Add info print showing which model is sampled in the current iteration
1 parent 8f03986 commit ab7ef24

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

openevolve/llm/ensemble.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ async def generate_with_context(
5757

5858
def _sample_model(self) -> LLMInterface:
5959
"""Sample a model from the ensemble based on weights"""
60+
logger.info(f"Sampled model: {vars(self.models[index])['model']}")
6061
index = self.random_state.choices(range(len(self.models)), weights=self.weights, k=1)[0]
6162
return self.models[index]
6263

0 commit comments

Comments
 (0)