fix code quality with ruff

aleph_message/models/__init__.py

@@ -5,15 +5,7 @@
 from hashlib import sha256
 from json import JSONDecodeError
 from pathlib import Path
-from typing import (
-    Any,
-    Dict,
-    List,
-    Literal,
-    Optional,
-    Type,
-    Union, TypeVar, cast,
-)
+from typing import Any, Dict, List, Literal, Optional, Type, TypeVar, Union, cast
 
 from pydantic import BaseModel, Extra, Field, validator
 from typing_extensions import TypeAlias
@@ -204,7 +196,9 @@ class BaseMessage(BaseModel):
     size: Optional[int] = Field(
         default=None, description="Size of the content"
     )  # Almost always present
-    time: datetime.datetime = Field(description="Unix timestamp or datetime when the message was published")
+    time: datetime.datetime = Field(
+        description="Unix timestamp or datetime when the message was published"
+    )
     item_type: ItemType = Field(description="Storage method used for the content")
     item_content: Optional[str] = Field(
         default=None,
@@ -343,7 +337,7 @@ class InstanceMessage(BaseMessage):
 ]
 
 
-T = TypeVar('T', bound=AlephMessage)
+T = TypeVar("T", bound=AlephMessage)
 
 AlephMessageType: TypeAlias = Type[T]
 

aleph_message/models/execution/abstract.py

@@ -5,13 +5,9 @@
 
 from pydantic import Field
 
-from .environment import (
-    FunctionEnvironment,
-    HostRequirements,
-    MachineResources,
-)
-from .volume import MachineVolume
 from ..abstract import BaseContent, HashableModel
+from .environment import FunctionEnvironment, HostRequirements, MachineResources
+from .volume import MachineVolume
 
 
 class BaseExecutableContent(HashableModel, BaseContent, ABC):

aleph_message/models/execution/environment.py

@@ -81,10 +81,10 @@ class CpuProperties(HashableModel):
     class Config:
         extra = Extra.forbid
 
-class HypervisorType(str, Enum):
-    qemu = 'qemu'
-    firecracker = 'firecracker'
 
+class HypervisorType(str, Enum):
+    qemu = "qemu"
+    firecracker = "firecracker"
 
 
 class FunctionEnvironment(HashableModel):

aleph_message/models/execution/instance.py

@@ -1,13 +1,11 @@
 from __future__ import annotations
 
-from typing import Any, Dict
-
 from pydantic import Field
 
 from aleph_message.models.abstract import HashableModel
-from aleph_message.models.item_hash import ItemHash
+
 from .abstract import BaseExecutableContent
-from .volume import VolumePersistence, PersistentVolumeSizeMib, ParentVolume
+from .volume import ParentVolume, PersistentVolumeSizeMib, VolumePersistence
 
 
 class RootfsVolume(HashableModel):
@@ -17,6 +15,7 @@ class RootfsVolume(HashableModel):
     The root file system of an instance is built as a copy of a reference image, named parent
     image. The user determines a custom size and persistence model.
     """
+
     parent: ParentVolume
     persistence: VolumePersistence
     # Use the same size constraint as persistent volumes for now
@@ -29,4 +28,3 @@ class InstanceContent(BaseExecutableContent):
     rootfs: RootfsVolume = Field(
         description="Root filesystem of the system, will be booted by the kernel"
     )
-

aleph_message/models/execution/program.py

@@ -4,11 +4,11 @@
 
 from pydantic import Field
 
-from .environment import FunctionTriggers
 from ..abstract import HashableModel
 from ..item_hash import ItemHash
 from .abstract import BaseExecutableContent
 from .base import Encoding, MachineType
+from .environment import FunctionTriggers
 
 
 class FunctionRuntime(HashableModel):

aleph_message/models/execution/volume.py

@@ -2,9 +2,9 @@
 
 from abc import ABC, abstractmethod
 from enum import Enum
-from typing import Literal, Optional, Type, Union
+from typing import Literal, Optional, Union
 
-from pydantic import ConstrainedInt, Extra, conint
+from pydantic import ConstrainedInt, Extra
 
 from ...utils import Gigabytes, gigabyte_to_mebibyte
 from ..abstract import HashableModel

aleph_message/tests/test_models.py

@@ -3,7 +3,6 @@
 from os import listdir
 from os.path import isdir, join
 from pathlib import Path
-from typing import Dict
 
 import pytest
 import requests
@@ -13,6 +12,7 @@
 from aleph_message.exceptions import UnknownHashError
 from aleph_message.models import (
     AggregateMessage,
+    AlephMessage,
     ForgetMessage,
     InstanceMessage,
     ItemType,
@@ -25,7 +25,7 @@
     create_message_from_file,
     create_message_from_json,
     create_new_message,
-    parse_message, AlephMessage,
+    parse_message,
 )
 from aleph_message.tests.download_messages import MESSAGES_STORAGE_PATH
 
@@ -289,10 +289,10 @@ def test_create_new_message():
     assert new_message_1
     assert new_message_1.type == MessageType.post
     # Check that the time was converted to a datetime
-    assert new_message_1.time.isoformat() == '2021-07-07T10:04:47.017000+00:00'
+    assert new_message_1.time.isoformat() == "2021-07-07T10:04:47.017000+00:00"
 
     # The time field can be either a float or a datetime as string
-    message_dict["time"] = '2021-07-07T10:04:47.017000+00:00'
+    message_dict["time"] = "2021-07-07T10:04:47.017000+00:00"
     new_message_2 = create_message_from_json(
         json.dumps(message_dict), factory=PostMessage
     )