From 4fa347242bd5de948cecb48c5375a5743b6d22f1 Mon Sep 17 00:00:00 2001 From: Paillat Date: Fri, 17 May 2024 11:26:45 +0200 Subject: [PATCH] :art: Run linter --- src/engines/AIImageEngine/A1111AIImageEngine.py | 2 +- src/engines/AIImageEngine/BaseAIImageEngine.py | 2 +- src/engines/AIImageEngine/DallEAIImageEngine.py | 2 +- src/engines/Pipelines/ScriptedVideoPipeline.py | 5 ++++- src/engines/StockImageEngine/BaseStockImageEngine.py | 2 +- src/engines/StockImageEngine/GoogleStockImageEngine.py | 4 +++- 6 files changed, 11 insertions(+), 6 deletions(-) diff --git a/src/engines/AIImageEngine/A1111AIImageEngine.py b/src/engines/AIImageEngine/A1111AIImageEngine.py index 8a3006e..4f1ffab 100644 --- a/src/engines/AIImageEngine/A1111AIImageEngine.py +++ b/src/engines/AIImageEngine/A1111AIImageEngine.py @@ -32,7 +32,7 @@ class A1111AIImageEngine(BaseAIImageEngine): super().__init__() - def generate(self, prompt: str, start: float, end: float, i= "") -> mp.ImageClip: + def generate(self, prompt: str, start: float, end: float, i="") -> mp.ImageClip: max_width = self.ctx.width / 3 * 2 try: url = self.base_url + "/sdapi/v1/txt2img" diff --git a/src/engines/AIImageEngine/BaseAIImageEngine.py b/src/engines/AIImageEngine/BaseAIImageEngine.py index a1adf90..170c999 100644 --- a/src/engines/AIImageEngine/BaseAIImageEngine.py +++ b/src/engines/AIImageEngine/BaseAIImageEngine.py @@ -11,7 +11,7 @@ class BaseAIImageEngine(BaseEngine): """ @abstractmethod - def generate(self, prompt: str, start: float, end: float, i = "") -> mp.ImageClip: + def generate(self, prompt: str, start: float, end: float, i="") -> mp.ImageClip: """ Ge """ diff --git a/src/engines/AIImageEngine/DallEAIImageEngine.py b/src/engines/AIImageEngine/DallEAIImageEngine.py index 74705fa..2af17a4 100644 --- a/src/engines/AIImageEngine/DallEAIImageEngine.py +++ b/src/engines/AIImageEngine/DallEAIImageEngine.py @@ -33,7 +33,7 @@ class DallEAIImageEngine(BaseAIImageEngine): super().__init__() - def generate(self, prompt: str, start: float, end: float, i = "") -> mp.ImageClip: + def generate(self, prompt: str, start: float, end: float, i="") -> mp.ImageClip: max_width = self.ctx.width / 3 * 2 size: Literal["1024x1024", "1024x1792", "1792x1024"] = ( "1024x1024" diff --git a/src/engines/Pipelines/ScriptedVideoPipeline.py b/src/engines/Pipelines/ScriptedVideoPipeline.py index d62f293..ed2bfd6 100644 --- a/src/engines/Pipelines/ScriptedVideoPipeline.py +++ b/src/engines/Pipelines/ScriptedVideoPipeline.py @@ -39,7 +39,10 @@ class ScriptedVideoPipeline(BasePipeline): def get_assets_concurrent(self, assets: list[dict[str, str]]) -> list[mp.VideoClip]: results = [] with ThreadPoolExecutor() as executor: - futures = [executor.submit(self.get_asset, asset, i) for i, asset in enumerate(assets)] + futures = [ + executor.submit(self.get_asset, asset, i) + for i, asset in enumerate(assets) + ] for future in as_completed(futures): try: results.append(future.result()) diff --git a/src/engines/StockImageEngine/BaseStockImageEngine.py b/src/engines/StockImageEngine/BaseStockImageEngine.py index 4aaae5d..1a6a373 100644 --- a/src/engines/StockImageEngine/BaseStockImageEngine.py +++ b/src/engines/StockImageEngine/BaseStockImageEngine.py @@ -11,7 +11,7 @@ class BaseStockImageEngine(BaseEngine): """ @abstractmethod - def get(self, query: str, start: float, end: float, i = "") -> mp.ImageClip: + def get(self, query: str, start: float, end: float, i="") -> mp.ImageClip: """ Get a stock image based on a query. diff --git a/src/engines/StockImageEngine/GoogleStockImageEngine.py b/src/engines/StockImageEngine/GoogleStockImageEngine.py index f2b16b7..0c09e22 100644 --- a/src/engines/StockImageEngine/GoogleStockImageEngine.py +++ b/src/engines/StockImageEngine/GoogleStockImageEngine.py @@ -42,7 +42,9 @@ class GoogleStockImageEngine(BaseStockImageEngine): custom_image_name="temp", ) # we find the file called temp. extension - filename = [f for f in os.listdir(f"./temp{i}/") if f.startswith("temp.")][0] + filename = [f for f in os.listdir(f"./temp{i}/") if f.startswith("temp.")][ + 0 + ] img = mp.ImageClip(f"./temp{i}/{filename}") # delete the temp folder except Exception as e: