Fix ruff & mypy errors (#2394)

2025-05-20 01:00:04 -07:00
parent 20fb787cc1
commit 5728b918b5
2 changed files with 262 additions and 214 deletions
--- a/skyvern/cli/commands.py
+++ b/skyvern/cli/commands.py
--- a/skyvern/library/skyvern.py
+++ b/skyvern/library/skyvern.py
@@ -300,7 +300,7 @@ class Skyvern(AsyncSkyvern):
        totp_url: str | None = None,
        title: str | None = None,
        error_code_mapping: dict[str, str] | None = None,
-        data_extraction_schema: dict[str, Any] | None = None,
+        data_extraction_schema: dict[str, Any] | str | None = None,
        proxy_location: ProxyLocation | None = None,
        max_steps: int | None = None,
        wait_for_completion: bool = True,