[qa] Fixed mypy violations

Author: Aryamanz29

tests/integration/aio/conftest.py

@@ -37,6 +37,7 @@ async def client():
     client = AsyncAtlanClient()
     yield client
 
+
 @pytest_asyncio.fixture(scope="module")
 async def token_client():
     """Async Atlan client fixture for api token integration tests."""

tests/integration/aio/test_client.py

@@ -87,18 +87,24 @@ async def token(token_client: AsyncAtlanClient) -> AsyncGenerator[ApiToken, None
 
 
 @pytest_asyncio.fixture(scope="module")
-async def expired_token(token_client: AsyncAtlanClient) -> AsyncGenerator[ApiToken, None]:
+async def expired_token(
+    token_client: AsyncAtlanClient,
+) -> AsyncGenerator[ApiToken, None]:
     token = None
     try:
-        token = await token_client.token.create(f"{MODULE_NAME}-expired", validity_seconds=1)
+        token = await token_client.token.create(
+            f"{MODULE_NAME}-expired", validity_seconds=1
+        )
         time.sleep(5)
         yield token
     finally:
         await delete_token_async(token_client, token)
 
 
 @pytest_asyncio.fixture(scope="module")
-async def argo_fake_token(token_client: AsyncAtlanClient) -> AsyncGenerator[ApiToken, None]:
+async def argo_fake_token(
+    token_client: AsyncAtlanClient,
+) -> AsyncGenerator[ApiToken, None]:
     token = None
     try:
         token = await token_client.token.create(f"{MODULE_NAME}-fake-argo")

tests/integration/client.py

@@ -32,6 +32,7 @@ def client() -> Generator[AtlanClient, None, None]:
 
     yield client
 
+
 @pytest.fixture(scope="module")
 def token_client() -> Generator[AtlanClient, None, None]:
     DEFAULT_RETRY.total = 0