fix: isort issues

annotation/annotation/annotations/resources.py

@@ -2,20 +2,33 @@
 from typing import Dict, List, Optional, Set
 from uuid import UUID
 
-from fastapi import APIRouter, Depends, HTTPException, Path, Query, Response, status
+from fastapi import (
+    APIRouter,
+    Depends,
+    HTTPException,
+    Path,
+    Query,
+    Response,
+    status,
+)
 from sqlalchemy import and_, desc
 from sqlalchemy.orm import Session
 from tenant_dependency import TenantData
 
-from annotation.categories.services import combine_categories, fetch_bunch_categories_db
+from annotation.categories.services import (
+    combine_categories,
+    fetch_bunch_categories_db,
+)
 from annotation.database import get_db
 from annotation.errors import NoSuchRevisionsError
 from annotation.jobs.services import update_jobs_categories
 from annotation.microservice_communication import jobs_communication
 from annotation.microservice_communication.assets_communication import (
     get_file_path_and_bucket,
 )
-from annotation.microservice_communication.search import X_CURRENT_TENANT_HEADER
+from annotation.microservice_communication.search import (
+    X_CURRENT_TENANT_HEADER,
+)
 from annotation.schemas import (
     AnnotatedDocSchema,
     BadRequestErrorSchema,

annotation/annotation/categories/resources.py

@@ -8,7 +8,9 @@
 from annotation.database import get_db
 from annotation.errors import NoSuchCategoryError
 from annotation.filters import CategoryFilter
-from annotation.microservice_communication.search import X_CURRENT_TENANT_HEADER
+from annotation.microservice_communication.search import (
+    X_CURRENT_TENANT_HEADER,
+)
 from annotation.schemas import (
     BadRequestErrorSchema,
     CategoryBaseSchema,

annotation/annotation/distribution/resources.py

@@ -6,7 +6,11 @@
 from tenant_dependency import TenantData
 
 from annotation.database import get_db
-from annotation.distribution import distribute, find_unassigned_files, prepare_response
+from annotation.distribution import (
+    distribute,
+    find_unassigned_files,
+    prepare_response,
+)
 from annotation.errors import FieldConstraintError
 from annotation.jobs import (
     check_annotators,
@@ -17,7 +21,9 @@
     get_files_info,
     prepare_files_for_distribution,
 )
-from annotation.microservice_communication.search import X_CURRENT_TENANT_HEADER
+from annotation.microservice_communication.search import (
+    X_CURRENT_TENANT_HEADER,
+)
 from annotation.models import File, Job, User
 from annotation.schemas import (
     BadRequestErrorSchema,

annotation/annotation/filters.py

@@ -1,6 +1,12 @@
 from filter_lib import create_filter_model
 
-from annotation.models import AnnotatedDoc, Category, Job, ManualAnnotationTask, User
+from annotation.models import (
+    AnnotatedDoc,
+    Category,
+    Job,
+    ManualAnnotationTask,
+    User,
+)
 
 ADDITIONAL_TASK_FIELDS = ["file_name", "job_name"]
 

annotation/annotation/jobs/resources.py

@@ -15,12 +15,16 @@
 from annotation.database import get_db
 from annotation.distribution import distribute, redistribute
 from annotation.filters import CategoryFilter
-from annotation.microservice_communication.assets_communication import get_files_info
+from annotation.microservice_communication.assets_communication import (
+    get_files_info,
+)
 from annotation.microservice_communication.jobs_communication import (
     JobUpdateException,
     update_job_status,
 )
-from annotation.microservice_communication.search import X_CURRENT_TENANT_HEADER
+from annotation.microservice_communication.search import (
+    X_CURRENT_TENANT_HEADER,
+)
 from annotation.schemas import (
     BadRequestErrorSchema,
     CategoryResponseSchema,
@@ -42,7 +46,14 @@
 from annotation.tags import FILES_TAG, JOBS_TAG
 from annotation.token_dependency import TOKEN
 
-from ..models import AnnotatedDoc, Category, File, Job, ManualAnnotationTask, User
+from ..models import (
+    AnnotatedDoc,
+    Category,
+    File,
+    Job,
+    ManualAnnotationTask,
+    User,
+)
 from .services import (
     add_users,
     collect_job_names,

annotation/annotation/jobs/services.py

@@ -12,9 +12,17 @@
 from annotation.categories import fetch_bunch_categories_db
 from annotation.categories.services import response_object_from_db
 from annotation.database import Base
-from annotation.errors import EnumValidationError, FieldConstraintError, WrongJobError
-from annotation.microservice_communication.assets_communication import get_files_info
-from annotation.microservice_communication.jobs_communication import get_job_names
+from annotation.errors import (
+    EnumValidationError,
+    FieldConstraintError,
+    WrongJobError,
+)
+from annotation.microservice_communication.assets_communication import (
+    get_files_info,
+)
+from annotation.microservice_communication.jobs_communication import (
+    get_job_names,
+)
 from annotation.models import (
     Category,
     File,

annotation/annotation/metadata/resources.py

@@ -1,6 +1,8 @@
 from fastapi import APIRouter, status
 
-from annotation.microservice_communication.search import X_CURRENT_TENANT_HEADER
+from annotation.microservice_communication.search import (
+    X_CURRENT_TENANT_HEADER,
+)
 from annotation.schemas import EntitiesStatusesSchema
 from annotation.tags import METADATA_TAG, TASKS_TAG
 

annotation/annotation/microservice_communication/task.py

@@ -11,7 +11,10 @@
     BEARER,
     HEADER_TENANT,
 )
-from annotation.schemas import AgreementScoreServiceInput, AgreementScoreServiceResponse
+from annotation.schemas import (
+    AgreementScoreServiceInput,
+    AgreementScoreServiceResponse,
+)
 
 load_dotenv(find_dotenv())
 AGREEMENT_SCORE_SERVICE_HOST = os.environ.get("AGREEMENT_SCORE_SERVICE_HOST")

annotation/annotation/revisions/resources.py

@@ -5,7 +5,9 @@
 from starlette import status
 
 from annotation.database import get_db
-from annotation.microservice_communication.search import X_CURRENT_TENANT_HEADER
+from annotation.microservice_communication.search import (
+    X_CURRENT_TENANT_HEADER,
+)
 from annotation.models import AnnotatedDoc
 from annotation.schemas import AnnotatedDocSchema, ConnectionErrorSchema
 from annotation.tags import ANNOTATION_TAG, REVISION_TAG

annotation/annotation/tasks/resources.py

@@ -76,7 +76,10 @@
     ValidationSchema,
 )
 from annotation.tags import REVISION_TAG, TASKS_TAG
-from annotation.tasks.validation import create_annotation_tasks, create_validation_tasks
+from annotation.tasks.validation import (
+    create_annotation_tasks,
+    create_validation_tasks,
+)
 from annotation.token_dependency import TOKEN
 
 from ..models import File, Job, ManualAnnotationTask

annotation/annotation/tasks/services.py

@@ -23,14 +23,20 @@
 )
 from annotation.errors import CheckFieldError, FieldConstraintError
 from annotation.filters import ADDITIONAL_TASK_FIELDS, TaskFilter
-from annotation.jobs import get_jobs_by_name, update_files, update_user_overall_load
+from annotation.jobs import (
+    get_jobs_by_name,
+    update_files,
+    update_user_overall_load,
+)
 from annotation.logger import Logger
 from annotation.microservice_communication.assets_communication import (
     get_file_names_by_request,
     get_file_path_and_bucket,
     get_files_by_request,
 )
-from annotation.microservice_communication.search import get_user_names_by_request
+from annotation.microservice_communication.search import (
+    get_user_names_by_request,
+)
 from annotation.microservice_communication.task import get_agreement_score
 from annotation.models import (
     AgreementMetrics,

annotation/tests/conftest.py

@@ -87,7 +87,13 @@
     TASKS_TEST_PROGRESS,
 )
 from tests.test_get_pages_info import PAGES_INFO_ENTITIES
-from tests.test_get_revisions import JOBS_IDS, PAGE, PAGES_PATHS, REVISIONS, USERS_IDS
+from tests.test_get_revisions import (
+    JOBS_IDS,
+    PAGE,
+    PAGES_PATHS,
+    REVISIONS,
+    USERS_IDS,
+)
 from tests.test_get_revisions_without_annotation import (
     REV_WITHOUT_ANNOTATION_DOC_1,
     REV_WITHOUT_ANNOTATION_DOC_2,
@@ -96,7 +102,10 @@
     REV_WITHOUT_ANNOTATION_TASK,
 )
 from tests.test_get_unassigned_files import UNASSIGNED_FILES_ENTITIES
-from tests.test_get_users_for_job import USERS_FOR_JOB_ANNOTATORS, USERS_FOR_JOB_JOBS
+from tests.test_get_users_for_job import (
+    USERS_FOR_JOB_ANNOTATORS,
+    USERS_FOR_JOB_JOBS,
+)
 from tests.test_job_categories import CATEGORIES_USERS, MOCK_ID
 from tests.test_post import POST_JOBS, TEST_POST_USERS
 from tests.test_post_annotation import (

annotation/tests/test_annotators_overall_load.py

@@ -11,7 +11,9 @@
 
 from annotation.jobs import update_user_overall_load
 from annotation.main import app
-from annotation.microservice_communication.assets_communication import ASSETS_FILES_URL
+from annotation.microservice_communication.assets_communication import (
+    ASSETS_FILES_URL,
+)
 from annotation.models import (
     AnnotatedDoc,
     Category,

annotation/tests/test_get_child_categories.py

@@ -6,19 +6,26 @@
 from fastapi.testclient import TestClient
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.orm import Session
+from tests.consts import CATEGORIES_PATH, POST_JOBS_PATH
+from tests.override_app_dependency import (
+    TEST_HEADERS,
+    TEST_TENANT,
+    TEST_TOKEN,
+    app,
+)
+from tests.test_job_categories import prepare_job_body
+from tests.test_post_next_task import ASSETS_RESPONSE
 
-from annotation.microservice_communication.assets_communication import ASSETS_FILES_URL
+from annotation.microservice_communication.assets_communication import (
+    ASSETS_FILES_URL,
+)
 from annotation.microservice_communication.search import (
     AUTHORIZATION,
     BEARER,
     HEADER_TENANT,
 )
 from annotation.models import Category
 from annotation.schemas import CategoryTypeSchema
-from tests.consts import CATEGORIES_PATH, POST_JOBS_PATH
-from tests.override_app_dependency import TEST_HEADERS, TEST_TENANT, TEST_TOKEN, app
-from tests.test_job_categories import prepare_job_body
-from tests.test_post_next_task import ASSETS_RESPONSE
 
 #  Cyclic categories have tree hierarchical structure of ids:
 #  "1" -> "2" -> "4" -> "1"

annotation/tests/test_get_pages_info.py

@@ -13,7 +13,13 @@
     BEARER,
     HEADER_TENANT,
 )
-from annotation.models import AnnotatedDoc, File, Job, ManualAnnotationTask, User
+from annotation.models import (
+    AnnotatedDoc,
+    File,
+    Job,
+    ManualAnnotationTask,
+    User,
+)
 from annotation.schemas import TaskStatusEnumSchema, ValidationSchema
 from annotation.tasks import get_task_revisions
 

annotation/tests/test_job_categories.py

@@ -6,7 +6,12 @@
 from fastapi.testclient import TestClient
 from sqlalchemy.orm import Session
 from tests.consts import POST_JOBS_PATH
-from tests.override_app_dependency import TEST_HEADERS, TEST_TENANT, TEST_TOKEN, app
+from tests.override_app_dependency import (
+    TEST_HEADERS,
+    TEST_TENANT,
+    TEST_TOKEN,
+    app,
+)
 from tests.test_category_crud import prepare_category_body
 
 from annotation.microservice_communication.search import (

annotation/tests/test_microservices_search.py

@@ -2,6 +2,7 @@
 import responses
 from fastapi import HTTPException
 from requests import ConnectionError, RequestException, Timeout
+from tests.override_app_dependency import TEST_HEADERS, TEST_TENANT, TEST_TOKEN
 
 from annotation.microservice_communication.search import (
     PAGE_SIZE,
@@ -11,8 +12,10 @@
     get_response,
 )
 from annotation.models import ManualAnnotationTask
-from annotation.schemas import ExpandedManualAnnotationTaskSchema, TaskStatusEnumSchema
-from tests.override_app_dependency import TEST_HEADERS, TEST_TENANT, TEST_TOKEN
+from annotation.schemas import (
+    ExpandedManualAnnotationTaskSchema,
+    TaskStatusEnumSchema,
+)
 
 AMOUNT_OF_ELEMENTS = 150
 

annotation/tests/test_post.py

@@ -8,7 +8,9 @@
 from sqlalchemy.orm import Session
 from tests.override_app_dependency import TEST_HEADERS, TEST_TENANT, app
 
-from annotation.microservice_communication.assets_communication import ASSETS_URL
+from annotation.microservice_communication.assets_communication import (
+    ASSETS_URL,
+)
 from annotation.models import Category, File, Job, ManualAnnotationTask, User
 from annotation.schemas import CategoryTypeSchema, ValidationSchema
 

annotation/tests/test_post_annotation.py

@@ -12,6 +12,14 @@
 from requests import RequestException
 from sqlalchemy.exc import DBAPIError, SQLAlchemyError
 from sqlalchemy.orm import Session
+from tests.consts import ANNOTATION_PATH
+from tests.override_app_dependency import (
+    TEST_HEADERS,
+    TEST_TENANT,
+    TEST_TOKEN,
+    app,
+)
+from tests.test_tasks_crud_ud import construct_path
 
 from annotation.annotations import (
     MANIFEST,
@@ -27,7 +35,9 @@
     upload_pages_to_minio,
 )
 from annotation.kafka_client import producers
-from annotation.microservice_communication.assets_communication import ASSETS_FILES_URL
+from annotation.microservice_communication.assets_communication import (
+    ASSETS_FILES_URL,
+)
 from annotation.microservice_communication.search import (
     AUTHORIZATION,
     BEARER,
@@ -49,9 +59,6 @@
     TaskStatusEnumSchema,
     ValidationSchema,
 )
-from tests.consts import ANNOTATION_PATH
-from tests.override_app_dependency import TEST_HEADERS, TEST_TENANT, TEST_TOKEN, app
-from tests.test_tasks_crud_ud import construct_path
 
 client = TestClient(app)