[Fix] Deepfake detection signature tests

edenai · Dec 16, 2024 · a4052a3 · a4052a3
1 parent 2d1da56
commit a4052a3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/edenai_apis/features/image/image_interface.py b/edenai_apis/features/image/image_interface.py
@@ -66,6 +66,9 @@
 from edenai_apis.features.image.landmark_detection.landmark_detection_dataclass import (
     LandmarkDetectionDataClass,
 )
+from edenai_apis.features.image.deepfake_detection.deepfake_detection_dataclass import (
+    DeepfakeDetectionDataClass,
+)
 from edenai_apis.features.image.logo_detection.logo_detection_dataclass import (
     LogoDetectionDataClass,
 )
@@ -181,7 +184,7 @@ def image__landmark_detection(
     @abstractmethod
     def image__deepfake_detection(
         self, file: str, file_url: str = ""
-    ) -> ResponseType[LandmarkDetectionDataClass]:
+    ) -> ResponseType[DeepfakeDetectionDataClass]:
         """
         Detect altered images via inconsistencies.
         (eg: popular tourist spots like Eiffel Tower)

diff --git a/edenai_apis/features/video/video_interface.py b/edenai_apis/features/video/video_interface.py
@@ -12,6 +12,9 @@
     TextDetectionAsyncDataClass,
     QuestionAnswerDataClass,
 )
+from edenai_apis.features.video.deepfake_detection_async.deepfake_detection_async_dataclass import (
+    DeepfakeDetectionAsyncDataClass,
+)
 from edenai_apis.utils.types import AsyncBaseResponseType, AsyncLaunchJobResponseType
 
 
@@ -55,7 +58,7 @@ def video__deepfake_detection_async__launch_job(
     @abstractmethod
     def video__deepfake_detection_async__get_job_result(
         self, provider_job_id: str
-    ) -> AsyncBaseResponseType[ExplicitContentDetectionAsyncDataClass]:
+    ) -> AsyncBaseResponseType[DeepfakeDetectionAsyncDataClass]:
         """Get the result of an asynchronous job by its ID
         Args:
             - provider_job_id (str): id of async job