diff --git a/compass_sdk/compass.py b/compass_sdk/compass.py index afabc7d..b91e583 100644 --- a/compass_sdk/compass.py +++ b/compass_sdk/compass.py @@ -6,6 +6,12 @@ from typing import Dict, Iterator, List, Optional, Tuple, Union import requests +from joblib import Parallel, delayed +from pydantic import BaseModel +from requests.exceptions import InvalidSchema +from tenacity import RetryError, retry, retry_if_not_exception_type, stop_after_attempt, wait_fixed +from tqdm import tqdm + from compass_sdk import ( BatchPutDocumentsInput, Chunk, @@ -25,11 +31,6 @@ DEFAULT_MAX_RETRIES, DEFAULT_SLEEP_RETRY_SECONDS, ) -from joblib import Parallel, delayed -from pydantic import BaseModel -from requests.exceptions import InvalidSchema -from tenacity import RetryError, retry, retry_if_not_exception_type, stop_after_attempt, wait_fixed -from tqdm import tqdm @dataclass diff --git a/compass_sdk/parser.py b/compass_sdk/parser.py index 9dada44..e020305 100644 --- a/compass_sdk/parser.py +++ b/compass_sdk/parser.py @@ -4,6 +4,7 @@ from typing import Any, Callable, Dict, Iterable, List, Optional, Union import requests + from compass_sdk import ( BatchProcessFilesParameters, CompassDocument,