From a2c6823e8906a460fb68ac92a8a6ba62b6a1db6e Mon Sep 17 00:00:00 2001 From: Paillat Date: Sun, 21 Apr 2024 21:52:14 +0200 Subject: [PATCH] :label: Add types and function signatures --- src/engines/CaptioningEngine/BaseCaptioningEngine.py | 4 ++-- src/engines/UploadEngine/BaseUploadEngine.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/src/engines/CaptioningEngine/BaseCaptioningEngine.py b/src/engines/CaptioningEngine/BaseCaptioningEngine.py index be80b92..891f999 100644 --- a/src/engines/CaptioningEngine/BaseCaptioningEngine.py +++ b/src/engines/CaptioningEngine/BaseCaptioningEngine.py @@ -1,3 +1,4 @@ +import moviepy as mp from abc import abstractmethod from ..BaseEngine import BaseEngine @@ -5,5 +6,4 @@ from ..BaseEngine import BaseEngine class BaseCaptioningEngine(BaseEngine): @abstractmethod - def get_captions(self) -> None: - ... + def get_captions(self, words: list[dict[str, str]] = None) -> list[mp.TextClip]: ... diff --git a/src/engines/UploadEngine/BaseUploadEngine.py b/src/engines/UploadEngine/BaseUploadEngine.py index d5de7b7..2fa354c 100644 --- a/src/engines/UploadEngine/BaseUploadEngine.py +++ b/src/engines/UploadEngine/BaseUploadEngine.py @@ -9,5 +9,4 @@ class BaseUploadEngine(BaseEngine): ... @abstractmethod - def upload(self): - ... + def upload(self, title: str, description: str, path: str): ...