ENCOA-311

ielts_be/controllers/abc/exam/writing.py

@@ -1,4 +1,5 @@
 from abc import ABC, abstractmethod
+from typing import List
 
 from fastapi.datastructures import UploadFile
 
@@ -6,9 +7,9 @@ from fastapi.datastructures import UploadFile
 class IWritingController(ABC):
 
     @abstractmethod
-    async def get_writing_task_general_question(self, task: int, topic: str, difficulty: str):
+    async def get_writing_task_general_question(self, task: int, topic: str, difficulty: List[str]):
         pass
 
     @abstractmethod
-    async def get_writing_task_academic_question(self, task: int, attachment: UploadFile, difficulty: str):
+    async def get_writing_task_academic_question(self, task: int, attachment: UploadFile, difficulty: List[str]):
         pass