Skip to content

Commit eca4bef

Browse files
committed
More style checker fix to add Union[<type>, None]
Signed-off-by: mmelqin <[email protected]>
1 parent 505796f commit eca4bef

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

monai/deploy/operators/dicom_text_sr_writer_operator.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
import logging
1414
from pathlib import Path
1515
from random import randint
16-
from typing import Dict, List, Text
16+
from typing import Dict, List, Text, Union
1717

1818
from monai.deploy.utils.importutil import optional_import
1919
from monai.deploy.utils.version import get_sdk_semver
@@ -98,8 +98,8 @@ def __init__(
9898
self,
9999
copy_tags: bool,
100100
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,
103103
):
104104
"""Class to write DICOM SR SOP Instance for AI textual result in memeory or in a file.
105105
@@ -186,7 +186,7 @@ def compute(self, op_input: InputContext, op_output: OutputContext, context: Exe
186186
# Now ready to starting writing the DICOM insance
187187
self.write(result_text, dicom_series, output_dir)
188188

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):
190190
"""Writes DICOM object
191191
192192
Args:
@@ -246,7 +246,7 @@ def write(self, content_text, dicom_series: DICOMSeries, output_dir: Path):
246246
self.save_dcm_file(ds, file_path)
247247

248248
@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):
250250
logging.debug(f"DICOM dataset to be written:{data_set}")
251251

252252
# Write out the DCM file
@@ -264,8 +264,8 @@ def write_common_modules(
264264
copy_tags: bool,
265265
modality_type: str,
266266
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,
269269
):
270270
"""Writes DICOM object common modules with or without a reference DCIOM Series
271271

0 commit comments

Comments
 (0)