<jx3>[feat]fix missing argument

HornCopper · Aug 15, 2024 · c1c5a31 · c1c5a31
1 parent 8ede0fa
commit c1c5a31
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/tools/generate.py b/src/tools/generate.py
@@ -105,7 +105,8 @@ async def generate(
     viewport: dict = None,
     full_screen: bool = False,
     hide_classes: list = None,
-    device_scale_factor: float = 1.0
+    device_scale_factor: float = 1.0,
+    output: str = None
 ):
-    generator = ScreenshotGenerator(path, web, locate, first, delay, additional_css, viewport, full_screen, hide_classes, device_scale_factor)
+    generator = ScreenshotGenerator(path, web, locate, first, delay, additional_css, viewport, full_screen, hide_classes, device_scale_factor, output)
     return await generator.generate()