Spaces:
Restarting
Restarting
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(
|
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):
|