From 59fa78e642419a66b1aa6ce79844c55bdc70f5d7 Mon Sep 17 00:00:00 2001 From: Kerem Yilmaz Date: Mon, 27 May 2024 00:17:41 -0700 Subject: [PATCH] Move all cloud imports after skyvern imports (#371) Co-authored-by: Shuchang Zheng --- pyproject.toml | 4 +++- scripts/create_organization.py | 3 ++- streamlit_app/visualizer/streamlit.py | 1 + 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index a9a32a4ff..3c5cf5962 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -125,7 +125,9 @@ force_grid_wrap = 0 use_parentheses = true ensure_newline_before_comments = true line_length = 120 -sections = ["FUTURE", "STDLIB", "THIRDPARTY", "FIRSTPARTY", "LOCALFOLDER"] +known_skyvern = "skyvern" +known_cloud = "cloud" +sections = ["FUTURE", "STDLIB", "THIRDPARTY", "SKYVERN", "CLOUD", "FIRSTPARTY", "LOCALFOLDER"] skip_glob = [".tox", "__pycache__", "*.pyc", "venv*/*", "reports", "venv", "env", "node_modules", ".env", ".venv", "dist"] skip = ["webeye/actions/__init__.py", "forge/sdk/__init__.py"] diff --git a/scripts/create_organization.py b/scripts/create_organization.py index af1c6e46f..1060dd690 100644 --- a/scripts/create_organization.py +++ b/scripts/create_organization.py @@ -3,9 +3,10 @@ import typer -from scripts.create_api_key import create_org_api_token from skyvern.forge.app import DATABASE +from scripts.create_api_key import create_org_api_token + async def create_org(org_name: str, webhook_callback_url: str | None = None) -> None: organization = await DATABASE.create_organization(org_name, webhook_callback_url) diff --git a/streamlit_app/visualizer/streamlit.py b/streamlit_app/visualizer/streamlit.py index 1bf8fd2c1..d241b179b 100644 --- a/streamlit_app/visualizer/streamlit.py +++ b/streamlit_app/visualizer/streamlit.py @@ -7,6 +7,7 @@ from skyvern import analytics from skyvern.forge.sdk.schemas.tasks import ProxyLocation, TaskRequest + from streamlit_app.visualizer import styles from streamlit_app.visualizer.api import SkyvernClient from streamlit_app.visualizer.artifact_loader import (