Refactor operator imports and enhance code formatting

- Updated import statements across multiple operator files for improved organization and readability.
- Cleaned up whitespace and added missing imports in various files, including ImageOverlayWriter and JSONResultsWriter.
- Ensured consistent formatting and indentation in the ImageDirectoryLoader and test files.

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

Author: mocsharp

monai/deploy/operators/__init__.py

@@ -68,22 +68,22 @@
     ModelInfo,
 )
 from .image_directory_loader_operator import ImageDirectoryLoader
+from .image_overlay_writer_operator import ImageOverlayWriter
 from .inference_operator import InferenceOperator
 from .json_results_writer_operator import JSONResultsWriter
+from .llama3_vila_inference_operator import Llama3VILAInferenceOperator
 from .monai_bundle_inference_operator import (
     BundleConfigNames,
     IOMapping,
     MonaiBundleInferenceOperator,
 )
 from .monai_classification_operator import MonaiClassificationOperator
 from .monai_seg_inference_operator import InfererType, MonaiSegInferenceOperator
-from .nii_data_loader_operator import NiftiDataLoader
 from .nifti_directory_loader_operator import NiftiDirectoryLoader
 from .nifti_writer_operator import NiftiWriter
+from .nii_data_loader_operator import NiftiDataLoader
 from .png_converter_operator import PNGConverterOperator
+from .prompts_loader_operator import PromptsLoaderOperator
 from .publisher_operator import PublisherOperator
 from .stl_conversion_operator import STLConversionOperator, STLConverter
-from .image_overlay_writer_operator import ImageOverlayWriter
-from .prompts_loader_operator import PromptsLoaderOperator
-from .llama3_vila_inference_operator import Llama3VILAInferenceOperator
 from .vlm_results_writer_operator import VLMResultsWriterOperator

monai/deploy/operators/image_directory_loader_operator.py

@@ -142,6 +142,7 @@ def compute(self, op_input, op_output, context):
 def test():
     """Test the ImageDirectoryLoader operator."""
     import tempfile
+
     from PIL import Image as PILImageCreate
 
     # Create a temporary directory with test images

monai/deploy/operators/image_overlay_writer_operator.py

@@ -21,9 +21,9 @@
 - filename: base name (stem) for output file
 """
 
+import logging
 from pathlib import Path
 from typing import Optional, Tuple
-import logging
 
 import numpy as np
 

monai/deploy/operators/json_results_writer_operator.py

@@ -180,6 +180,7 @@ def _print_classification_summary(self, result: Dict[str, Any]):
 def test():
     """Test the JSONResultsWriter operator."""
     import tempfile
+
     import numpy as np
 
     with tempfile.TemporaryDirectory() as temp_dir:

monai/deploy/operators/monai_classification_operator.py

@@ -14,11 +14,11 @@
 from typing import List, Optional, Union
 
 import torch
-from monai.bundle import ConfigParser
-from monai.transforms import Compose
 
+from monai.bundle import ConfigParser
 from monai.deploy.core import AppContext, Fragment, Image, Operator, OperatorSpec
 from monai.deploy.utils.importutil import optional_import
+from monai.transforms import Compose
 
 # Dynamic class imports to match MONAI model loader behavior
 monai, _ = optional_import("monai")

tests/unit/test_vlm_operators.py

@@ -452,10 +452,10 @@ def tearDown(self):
     @patch("monai.deploy.operators.llama3_vila_inference_operator.AutoConfig")
     def test_end_to_end_flow(self, mock_autoconfig, mock_pil):
         """Test end-to-end flow of VLM operators."""
-        from monai.deploy.operators.prompts_loader_operator import PromptsLoaderOperator
         from monai.deploy.operators.llama3_vila_inference_operator import (
             Llama3VILAInferenceOperator,
         )
+        from monai.deploy.operators.prompts_loader_operator import PromptsLoaderOperator
         from monai.deploy.operators.vlm_results_writer_operator import (
             VLMResultsWriterOperator,
         )