13
13
import logging
14
14
from pathlib import Path
15
15
from random import randint
16
- from typing import Dict , List , Text
16
+ from typing import Dict , List , Text , Union
17
17
18
18
from monai .deploy .utils .importutil import optional_import
19
19
from monai .deploy .utils .version import get_sdk_semver
@@ -98,8 +98,8 @@ def __init__(
98
98
self ,
99
99
copy_tags : bool ,
100
100
model_info : ModelInfo ,
101
- equipment_info : EquipmentInfo = None ,
102
- custom_tags : Dict [str , str ] = None ,
101
+ equipment_info : Union [ EquipmentInfo , None ] = None ,
102
+ custom_tags : Union [ Dict [str , str ], None ] = None ,
103
103
):
104
104
"""Class to write DICOM SR SOP Instance for AI textual result in memeory or in a file.
105
105
@@ -186,7 +186,7 @@ def compute(self, op_input: InputContext, op_output: OutputContext, context: Exe
186
186
# Now ready to starting writing the DICOM insance
187
187
self .write (result_text , dicom_series , output_dir )
188
188
189
- def write (self , content_text , dicom_series : DICOMSeries , output_dir : Path ):
189
+ def write (self , content_text , dicom_series : Union [ DICOMSeries , None ] , output_dir : Path ):
190
190
"""Writes DICOM object
191
191
192
192
Args:
@@ -246,7 +246,7 @@ def write(self, content_text, dicom_series: DICOMSeries, output_dir: Path):
246
246
self .save_dcm_file (ds , file_path )
247
247
248
248
@staticmethod
249
- def save_dcm_file (data_set : Dataset , file_path : Path , validate_readable : bool = True ):
249
+ def save_dcm_file (data_set , file_path : Path , validate_readable : bool = True ):
250
250
logging .debug (f"DICOM dataset to be written:{ data_set } " )
251
251
252
252
# Write out the DCM file
@@ -264,8 +264,8 @@ def write_common_modules(
264
264
copy_tags : bool ,
265
265
modality_type : str ,
266
266
sop_class_uid : str ,
267
- model_info : ModelInfo = None ,
268
- equipment_info : EquipmentInfo = None ,
267
+ model_info : Union [ ModelInfo , None ] = None ,
268
+ equipment_info : Union [ EquipmentInfo , None ] = None ,
269
269
):
270
270
"""Writes DICOM object common modules with or without a reference DCIOM Series
271
271
0 commit comments