fix failing tests

This commit is contained in:
Brandon Hancock
2025-01-28 14:25:10 -05:00
parent a690bb91b3
commit 10b42bd0d4
2 changed files with 21 additions and 14 deletions

View File

@@ -492,6 +492,7 @@ class Crew(BaseModel):
train_crew = self.copy()
train_crew._setup_for_training(filename)
try:
for n_iteration in range(n_iterations):
train_crew._train_iteration = n_iteration
train_crew.kickoff(inputs=inputs)
@@ -503,10 +504,17 @@ class Crew(BaseModel):
result = TaskEvaluator(agent).evaluate_training_data(
training_data=training_data, agent_id=str(agent.id)
)
CrewTrainingHandler(filename).save_trained_data(
agent_id=str(agent.role), trained_data=result.model_dump()
)
except Exception as e:
self._logger.log("error", f"Training failed: {e}", color="red")
CrewTrainingHandler(TRAINING_DATA_FILE).clear()
CrewTrainingHandler(filename).clear()
raise
finally:
CrewTrainingHandler(TRAINING_DATA_FILE).close()
CrewTrainingHandler(filename).close()
def kickoff(
self,

View File

@@ -112,7 +112,6 @@ class TaskEvaluator:
"Cannot proceed with evaluation.\n"
"Please check your training implementation."
)
self._logger.log("critical", error_msg, color="red")
raise ValueError(error_msg)
final_aggregated_data += (