Apply suggestions from code review

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Signed-off-by: Adam Poulemanos <89049923+bashandbone@users.noreply.github.com>

Author: bashandbone

src/codeweaver/providers/config/categories/embedding.py

@@ -223,7 +223,7 @@ def client(self) -> LiteralSDKClient:
         with contextlib.suppress(AttributeError, KeyError, ValueError):
             is_sdkclient_member = SDKClient.from_string(self.provider.variable) is not None
         if self.provider.only_uses_own_client and is_sdkclient_member:
-            return cast("LiteralSDKClient", SDKClient.from_string(self.provider.variable))
+            return cast(LiteralSDKClient, SDKClient.from_string(self.provider.variable))
         if self.provider not in (Provider.AZURE, Provider.HEROKU):
             raise ValueError(
                 f"Cannot resolve embedding client for provider {self.provider.variable}.",
@@ -834,7 +834,7 @@ def validate_model_compatibility(self) -> AsymmetricEmbeddingProviderSettings:
     @property
     def dimension(self) -> NonNegativeInt:
         """Get the embedding dimension for both models (they must match)."""
-        return cast("NonNegativeInt", self.dimension_tuple[0] or 0)
+        return cast(NonNegativeInt, self.dimension_tuple[0] or 0)
 
     @property
     def datatype(self) -> str:

src/codeweaver/providers/config/categories/reranking.py

@@ -98,7 +98,7 @@ class BaseRerankingProviderSettings(BaseProviderCategorySettings):
     @property
     def client(self) -> LiteralSDKClient:
         """Return the reranking SDKClient enum member."""
-        return cast("LiteralSDKClient", SDKClient.from_string(self.provider.variable))
+        return cast(LiteralSDKClient, SDKClient.from_string(self.provider.variable))
 
     def is_cloud(self) -> bool:
         """Return True if the provider is a cloud provider, False otherwise."""

tests/integration/test_config_validation_flow.py

@@ -16,7 +16,7 @@
 
 from datetime import timedelta
 from pathlib import Path
-from unittest.mock import AsyncMock, Mock, NonCallableMock
+from unittest.mock import AsyncMock, Mock
 
 import pytest
 
@@ -115,8 +115,10 @@ def mock_checkpoint_manager(test_checkpoint_data: dict) -> Mock:
     manager.delete = AsyncMock()
     # Some tests may call `load_checkpoint()` instead of `load()`;
     # make it an alias so both return the same checkpoint object.
-    manager.load_checkpoint = manager.load
-    manager.validate_checkpoint_compatibility = AsyncMock(return_value=(True, "NONE"))
+    if hasattr(manager, "load_checkpoint"):
+        manager.load_checkpoint = manager.load
+    if hasattr(manager, "validate_checkpoint_compatibility"):
+        manager.validate_checkpoint_compatibility = AsyncMock(return_value=(True, "NONE"))
 
     # Configure synchronous fingerprint helpers with real CheckpointSettingsFingerprint
     # instances so ConfigChangeAnalyzer's ``is_compatible_with`` comparison works correctly.