ecemumutlu commited on
Commit
7820d13
·
1 Parent(s): c88093d

Fix task name error

Browse files
src/deepeval/deepeval_task_manager.py CHANGED
@@ -15,7 +15,7 @@ class Task(Enum):
15
  class DeepEvalTaskManager:
16
  def __init__(self, model_name, tasks: List[str]):
17
  self.model_name = model_name
18
- self.available_tasks = {task.name: getattr(self, task.name.lower()) for task in Task}
19
  self.tasks_to_run = self.validate_tasks(tasks)
20
 
21
  def validate_tasks(self, user_tasks):
 
15
  class DeepEvalTaskManager:
16
  def __init__(self, model_name, tasks: List[str]):
17
  self.model_name = model_name
18
+ self.available_tasks = {task.name: getattr(Task, task.name).value for task in Task}
19
  self.tasks_to_run = self.validate_tasks(tasks)
20
 
21
  def validate_tasks(self, user_tasks):