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

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(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):
@@ -33,8 +33,6 @@ class DeepEvalTaskManager:
33
  results = {}
34
  for task_name, task_method in self.tasks_to_run.items():
35
  task_enum = getattr(Task, task_name)
36
-
37
- # Get the value of the enum member
38
  task_value = task_enum.value
39
  results[task_value] = task_method() # Call the stored method reference
40
 
 
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.value) for task in Task}
19
  self.tasks_to_run = self.validate_tasks(tasks)
20
 
21
  def validate_tasks(self, user_tasks):
 
33
  results = {}
34
  for task_name, task_method in self.tasks_to_run.items():
35
  task_enum = getattr(Task, task_name)
 
 
36
  task_value = task_enum.value
37
  results[task_value] = task_method() # Call the stored method reference
38