Enhance type hinting and improve code clarity across operators

- Added type hints for lists and dictionaries in various operator classes to improve code readability and maintainability.
- Updated import statements to include type ignoring for YAML to prevent type checking issues.
- Refined the initialization of file lists in the GenericDirectoryScanner operator for better type safety.

Signed-off-by: Victor Chang <[email protected]>

Author: mocsharp

tools/pipeline-generator/pipeline_generator/config/settings.py

@@ -14,7 +14,7 @@
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Union
 
-import yaml
+import yaml  # type: ignore
 from pydantic import BaseModel, Field
 
 
@@ -130,6 +130,7 @@ def load_config(config_path: Optional[Path] = None) -> Settings:
                 model_id=None,
                 base_url="https://huggingface.co",
                 description="Official MONAI organization models",
+                model_type=None,
             )
         ]
     )

tools/pipeline-generator/pipeline_generator/templates/operators/generic_directory_scanner_operator.py

@@ -56,7 +56,7 @@ def __init__(
         self._case_sensitive = bool(case_sensitive)
 
         # State tracking
-        self._files = []
+        self._files: list[Path] = []
         self._current_index = 0
 
         super().__init__(fragment, *args, **kwargs)

tools/pipeline-generator/pipeline_generator/templates/operators/image_overlay_writer_operator.py

@@ -74,7 +74,7 @@ def compute(self, op_input, op_output, context):
 
     def _to_hwc_uint8(self, image) -> np.ndarray:
         if isinstance(image, Image):
-            arr = image.asnumpy()
+            arr: np.ndarray = image.asnumpy()
         else:
             arr = np.asarray(image)
         if arr.ndim != 3 or arr.shape[2] not in (3, 4):
@@ -89,7 +89,7 @@ def _to_hwc_uint8(self, image) -> np.ndarray:
 
     def _to_mask_uint8(self, pred) -> np.ndarray:
         if isinstance(pred, Image):
-            arr = pred.asnumpy()
+            arr: np.ndarray = pred.asnumpy()
         else:
             arr = np.asarray(pred)
         arr = np.squeeze(arr)

tools/pipeline-generator/pipeline_generator/templates/operators/json_results_writer_operator.py

@@ -97,7 +97,7 @@ def compute(self, op_input, op_output, context):
 
     def _process_prediction(self, pred: Any, filename: str) -> Dict[str, Any]:
         """Process various prediction formats into a JSON-serializable dictionary."""
-        result = {"filename": filename}
+        result: Dict[str, Any] = {"filename": filename}
 
         # Handle dictionary predictions (e.g., from MonaiBundleInferenceOperator)
         if isinstance(pred, dict):

tools/pipeline-generator/pipeline_generator/templates/operators/llama3_vila_inference_operator.py

@@ -224,7 +224,7 @@ def _create_image_overlay(self, image: Image, text: str) -> Image:
         # Break text into lines for better display
         words = text.split()
         lines = []
-        current_line = []
+        current_line: list[str] = []
         max_width = pil_image.width - 20  # Leave margin
 
         # Simple text wrapping (in production, use proper text metrics)

tools/pipeline-generator/pipeline_generator/templates/operators/prompts_loader_operator.py

@@ -15,7 +15,7 @@
 from typing import Any, Dict, List, Optional
 
 import numpy as np
-import yaml
+import yaml  # type: ignore
 
 from monai.deploy.core import Fragment, Image, Operator, OperatorSpec
 from monai.deploy.utils.importutil import optional_import