Fix: Linting tests did not pass:

Author: Antonyjin

src/aleph/sdk/conf.py

@@ -5,11 +5,10 @@
 from shutil import which
 from typing import ClassVar, Dict, Optional, Union
 
-from pydantic_settings import BaseSettings
-
 from aleph_message.models import Chain
 from aleph_message.models.execution.environment import HypervisorType
 from pydantic import BaseModel, ConfigDict, Field
+from pydantic_settings import BaseSettings
 
 from aleph.sdk.types import ChainInfo
 
@@ -142,12 +141,10 @@ class Settings(BaseSettings):
     DNS_PROGRAM_DOMAIN: ClassVar[str] = "program.public.aleph.sh"
     DNS_INSTANCE_DOMAIN: ClassVar[str] = "instance.public.aleph.sh"
     DNS_STATIC_DOMAIN: ClassVar[str] = "static.public.aleph.sh"
-    DNS_RESOLVERS: ClassVar[str] = ["9.9.9.9", "1.1.1.1"]
+    DNS_RESOLVERS: ClassVar[list[str]] = ["9.9.9.9", "1.1.1.1"]
 
     model_config = ConfigDict(
-       env_prefix="ALEPH_",
-       case_sensitive=False,
-       env_file=".env"
+        env_prefix="ALEPH_", case_sensitive=False, env_file=".env"
     )
 
 

src/aleph/sdk/utils.py

@@ -28,15 +28,13 @@
 from uuid import UUID
 from zipfile import BadZipFile, ZipFile
 
-from pydantic import BaseModel
-
+import pydantic_core
 from aleph_message.models import ItemHash, MessageType
 from aleph_message.models.execution.program import Encoding
 from aleph_message.models.execution.volume import MachineVolume
 from cryptography.hazmat.backends import default_backend
 from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
 from jwcrypto.jwa import JWA
-import pydantic_core
 
 from aleph.sdk.conf import settings
 from aleph.sdk.types import GenericMessage, SEVInfo, SEVMeasurement

tests/unit/aleph_vm_authentication.py

@@ -13,7 +13,7 @@
 from eth_account.messages import encode_defunct
 from jwcrypto import jwk
 from jwcrypto.jwa import JWA
-from pydantic import BaseModel, ValidationError, model_validator, field_validator
+from pydantic import BaseModel, ValidationError, field_validator, model_validator
 
 from aleph.sdk.utils import bytes_from_hex
 
@@ -74,7 +74,7 @@ def payload_must_be_hex(cls, value: bytes) -> bytes:
         return bytes_from_hex(value.decode())
 
     @model_validator(mode="after")
-    def check_expiry(cls, values) -> Dict[str, bytes]:
+    def check_expiry(cls, values: "SignedPubKeyHeader") -> "SignedPubKeyHeader":
         """Check that the token has not expired"""
         payload: bytes = values.payload
         content = SignedPubKeyPayload.model_validate_json(payload)
@@ -86,8 +86,7 @@ def check_expiry(cls, values) -> Dict[str, bytes]:
         return values
 
     @model_validator(mode="after")
-    def check_signature(cls, values: Dict[str, bytes]) -> Dict[str, bytes]:
-        """Check that the signature is valid"""
+    def check_signature(cls, values: "SignedPubKeyHeader") -> "SignedPubKeyHeader":
         signature: bytes = values.signature
         payload: bytes = values.payload
         content = SignedPubKeyPayload.model_validate_json(payload)