style: fixed precommit checks

Author: Aryamanz29

tests/integration/agentic_test.py

@@ -9,12 +9,7 @@
     Skill,
     SkillArtifact,
 )
-from pyatlan.model.enums import (
-    CertificateStatus,
-    ContextLifecycleStatus,
-    EntityStatus,
-    FileType,
-)
+from pyatlan.model.enums import CertificateStatus, EntityStatus, FileType
 from tests.integration.client import TestId, delete_asset
 
 MODULE_NAME = TestId.make_unique("AGENTIC")
@@ -134,9 +129,7 @@ def test_update_skill_certificate(client: AtlanClient, skill: Skill):
 # ── Context tests ───────────────────────────────────────────────────────────
 
 
-def test_context_repository(
-    client: AtlanClient, context_repository: ContextRepository
-):
+def test_context_repository(client: AtlanClient, context_repository: ContextRepository):
     assert context_repository
     assert context_repository.guid
     assert context_repository.qualified_name

tests/unit/model/context_artifact_test.py

@@ -1,9 +1,8 @@
 import pytest
 
-from pyatlan.model.assets import ContextArtifact, ContextRepository
+from pyatlan.model.assets import ContextArtifact
 from pyatlan.model.enums import FileType
 
-
 CONTEXT_ARTIFACT_NAME = "my-test-context-artifact"
 CONTEXT_REPO_QUALIFIED_NAME = "default/context/abc123"
 
@@ -51,12 +50,8 @@ def test_creator():
     assert sut.type_name == "ContextArtifact"
     assert sut.file_type == FileType.TXT
     assert sut.qualified_name
-    assert sut.qualified_name.startswith(
-        f"{CONTEXT_REPO_QUALIFIED_NAME}/artifact/txt/"
-    )
-    assert (
-        sut.context_repository_qualified_name == CONTEXT_REPO_QUALIFIED_NAME
-    )
+    assert sut.qualified_name.startswith(f"{CONTEXT_REPO_QUALIFIED_NAME}/artifact/txt/")
+    assert sut.context_repository_qualified_name == CONTEXT_REPO_QUALIFIED_NAME
 
 
 def test_creator_with_yaml_file_type():
@@ -100,9 +95,7 @@ def test_creator_populates_context_repository_ref():
     )
 
     assert sut.context_repository is not None
-    assert (
-        sut.context_repository.qualified_name == CONTEXT_REPO_QUALIFIED_NAME
-    )
+    assert sut.context_repository.qualified_name == CONTEXT_REPO_QUALIFIED_NAME
 
 
 def test_creator_generates_unique_qualified_names():

tests/unit/model/context_repository_test.py

@@ -3,7 +3,6 @@
 from pyatlan.model.assets import ContextRepository
 from pyatlan.model.enums import ContextLifecycleStatus
 
-
 CONTEXT_REPO_NAME = "my-test-context-repo"
 
 
@@ -13,9 +12,7 @@
         (None, "name is required"),
     ],
 )
-def test_creator_with_missing_parameters_raise_value_error(
-    name: str, message: str
-):
+def test_creator_with_missing_parameters_raise_value_error(name: str, message: str):
     with pytest.raises(ValueError, match=message):
         ContextRepository.creator(name=name)
 

tests/unit/model/skill_artifact_test.py

@@ -1,9 +1,8 @@
 import pytest
 
-from pyatlan.model.assets import Skill, SkillArtifact
+from pyatlan.model.assets import SkillArtifact
 from pyatlan.model.enums import FileType
 
-
 SKILL_ARTIFACT_NAME = "my-test-artifact"
 SKILL_QUALIFIED_NAME = "default/skill/abc123"
 

tests/unit/model/skill_test.py

@@ -1,8 +1,6 @@
 import pytest
 
 from pyatlan.model.assets import Skill
-from pyatlan.model.enums import FileType
-
 
 SKILL_NAME = "my-test-skill"
 
@@ -13,9 +11,7 @@
         (None, "name is required"),
     ],
 )
-def test_creator_with_missing_parameters_raise_value_error(
-    name: str, message: str
-):
+def test_creator_with_missing_parameters_raise_value_error(name: str, message: str):
     with pytest.raises(ValueError, match=message):
         Skill.creator(name=name)