lint

Author: rohitvinnakota-codecov

api/internal/gen_ai/serializers.py

@@ -1,15 +1,6 @@
-from graphql_api.types.owner.owner import AI_FEATURES_GH_APP_ID
-from rest_framework.permissions import IsAuthenticated
-from rest_framework.response import Response
-from rest_framework.views import APIView
 from rest_framework import serializers
 
-from codecov_auth.models import GithubAppInstallation, Owner
-from shared.django_apps.core.models import Repository
-from shared.license import get_current_license
-
 
 class GenAIAuthSerializer(serializers.Serializer):
     is_valid = serializers.BooleanField()
     repos = serializers.ListField(child=serializers.CharField(), required=False)
-

api/internal/gen_ai/views.py

@@ -1,15 +1,15 @@
-from graphql_api.types.owner.owner import AI_FEATURES_GH_APP_ID
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
 from rest_framework.views import APIView
-from shared.license import get_current_license
-
-from .serializers import GenAIAuthSerializer, LicenseSerializer
 
 from codecov_auth.models import (
     GithubAppInstallation,
     Owner,
 )
+from graphql_api.types.owner.owner import AI_FEATURES_GH_APP_ID
+
+from .serializers import GenAIAuthSerializer
+
 
 class GenAIAuthView(APIView):
     permission_classes = [IsAuthenticated]
@@ -27,7 +27,7 @@ def get(self, request, *args, **kwargs):
             ).first()
 
         if ai_features_app_install and ai_features_app_install.repository_service_ids:
-                repos = ai_features_app_install.repository_service_ids
+            repos = ai_features_app_install.repository_service_ids
 
         data = {
             "is_valid": bool(ai_features_app_install),

api/internal/tests/test_gen_ai.py

@@ -1,9 +1,10 @@
 from rest_framework.reverse import reverse
 from rest_framework.test import APITestCase
 from shared.django_apps.core.tests.factories import OwnerFactory, RepositoryFactory
-from utils.test_utils import Client
+
 from codecov_auth.models import GithubAppInstallation
 from graphql_api.types.owner.owner import AI_FEATURES_GH_APP_ID
+from utils.test_utils import Client
 
 
 class GenAIAuthViewTests(APITestCase):

utils/test_utils.py

@@ -45,10 +45,10 @@ def app(self) -> str:
     migrate_to = None
 
     def setUp(self) -> None:
-        assert self.migrate_from and self.migrate_to, (
-            "TestCase '{}' must define migrate_from and migrate_to properties".format(
-                type(self).__name__
-            )
+        assert (
+            self.migrate_from and self.migrate_to
+        ), "TestCase '{}' must define migrate_from and migrate_to properties".format(
+            type(self).__name__
         )
         self.migrate_from = [(self.app, self.migrate_from)]
         self.migrate_to = [(self.app, self.migrate_to)]