Fix: Solve code quality issues.

src/aleph/vm/pool.py

@@ -5,6 +5,7 @@
 import logging
 from collections.abc import Iterable
 from datetime import datetime, timezone
+from typing import List
 
 from aleph_message.models import (
     Chain,
@@ -18,7 +19,7 @@
 from aleph.vm.controllers.firecracker.snapshot_manager import SnapshotManager
 from aleph.vm.network.hostnetwork import Network, make_ipv6_allocator
 from aleph.vm.orchestrator.metrics import get_execution_records
-from aleph.vm.resources import get_gpu_info
+from aleph.vm.resources import GpuProperties, get_gpu_info
 from aleph.vm.systemd import SystemDManager
 from aleph.vm.utils import get_message_executable_content
 from aleph.vm.vm_type import VmType

src/aleph/vm/resources.py

@@ -1,8 +1,8 @@
 import subprocess
 from enum import Enum
-from typing import Optional, List
+from typing import List, Optional
 
-from pydantic import BaseModel, Field, Extra
+from pydantic import BaseModel, Extra, Field
 
 
 class GpuDeviceClass(str, Enum):
@@ -70,6 +70,8 @@ def parse_gpu_device_info(line: str) -> Optional[GpuProperties]:
     if not is_gpu_device_class(device_class):
         return None
 
+    device_class = GpuDeviceClass(device_class)
+
     vendor, vendor_id = device_vendor.split(" [", maxsplit=1)
     vendor_id = vendor_id[:-1]
     vendor_name = get_vendor_name(vendor_id)
@@ -79,6 +81,7 @@ def parse_gpu_device_info(line: str) -> Optional[GpuProperties]:
     device_id = f"{vendor_id}:{model_id}"
 
     return GpuProperties(
+        pci_host=pci_host,
         vendor=vendor_name,
         device_name=device_name,
         device_class=device_class,