fix: Add missing generic type

Author: jopemachine

src/ai/backend/storage/services/artifacts/huggingface.py

@@ -708,7 +708,7 @@ async def _import_models_batch(reporter: ProgressReporter) -> DispatchResult:
 # Import Pipeline Steps
 
 
-class HuggingFaceDownloadStep(ImportStep):
+class HuggingFaceDownloadStep(ImportStep[None]):
     """Step to download files from HuggingFace"""
 
     def __init__(self, registry_configs: dict[str, HuggingfaceConfig]) -> None:
@@ -1010,7 +1010,7 @@ def create_huggingface_import_pipeline(
     artifact_verifier_ctx: ArtifactVerifierContext,
 ) -> ImportPipeline:
     """Create ImportPipeline for HuggingFace based on storage step mappings."""
-    steps: list[ImportStep] = []
+    steps: list[ImportStep[Any]] = []
 
     # Add steps based on what's present in storage_step_mappings
     if ArtifactStorageImportStep.DOWNLOAD in storage_step_mappings:

src/ai/backend/storage/services/artifacts/reservoir.py

@@ -385,7 +385,7 @@ async def _import_models_batch(reporter: ProgressReporter) -> DispatchResult:
 # Import Pipeline Steps
 
 
-class ReservoirDownloadStep(ImportStep):
+class ReservoirDownloadStep(ImportStep[None]):
     """Step to copy files from Reservoir (effectively direct copy to download storage)"""
 
     def __init__(
@@ -955,7 +955,7 @@ def create_reservoir_import_pipeline(
     artifact_verifier_ctx: ArtifactVerifierContext,
 ) -> ImportPipeline:
     """Create ImportPipeline for Reservoir based on storage step mappings."""
-    steps: list[ImportStep] = []
+    steps: list[ImportStep[Any]] = []
 
     # Add steps based on what's present in storage_step_mappings
     if ArtifactStorageImportStep.DOWNLOAD in storage_step_mappings: