Spaces:
Sleeping
Sleeping
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(
|
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 |
|