Skip to content

Commit 0db0311

Browse files
wylihjmjohnson
authored andcommitted
Silence pytype identified warnings
1 parent 1550926 commit 0db0311

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

monai/data/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ def worker_init_fn(worker_id):
234234
It can set different random seed for the transforms in different workers.
235235
236236
"""
237-
worker_info = torch.utils.data.get_worker_info()
237+
worker_info = torch.utils.data.get_worker_info() # type: ignore
238238
if hasattr(worker_info.dataset, "transform") and hasattr(worker_info.dataset.transform, "set_random_state"):
239239
worker_info.dataset.transform.set_random_state(worker_info.seed % (2 ** 32))
240240

monai/engines/workflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def __init__(
6868
self.logger.info("Will add AMP support when PyTorch v1.6 released.")
6969
if not isinstance(device, torch.device):
7070
raise ValueError("device must be PyTorch device object.")
71-
if not isinstance(data_loader, torch.utils.data.DataLoader):
71+
if not isinstance(data_loader, torch.utils.data.DataLoader): # type: ignore
7272
raise ValueError("data_loader must be PyTorch DataLoader.")
7373

7474
# set all sharable data for the workflow based on Ignite engine.state

monai/handlers/validation_handler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ class ValidationHandler:
2020
2121
"""
2222

23-
def __init__(self, validator: Evaluator, interval: int, epoch_level: bool = True) -> None:
23+
def __init__(self, validator: Evaluator, interval: int, epoch_level: bool = True) -> None: # type: ignore
2424
"""
2525
Args:
2626
validator (Evaluator): run the validator when trigger validation, suppose to be Evaluator.
@@ -29,7 +29,7 @@ def __init__(self, validator: Evaluator, interval: int, epoch_level: bool = True
2929
`True` is epoch level, `False` is iteration level.
3030
3131
"""
32-
if not isinstance(validator, Evaluator):
32+
if not isinstance(validator, Evaluator): # type: ignore
3333
raise ValueError("validator must be Evaluator ignite engine.")
3434
self.validator = validator
3535
self.interval = interval

monai/transforms/spatial/array.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,7 @@ def __init__(
410410

411411
if self.use_gpu:
412412
try:
413-
from cupyx.scipy.ndimage import zoom as zoom_gpu
413+
from cupyx.scipy.ndimage import zoom as zoom_gpu # type: ignore
414414

415415
self._zoom = zoom_gpu
416416
except ImportError:
@@ -434,7 +434,7 @@ def __call__(
434434
"""
435435
zoomed = list()
436436
if self.use_gpu:
437-
import cupy
437+
import cupy # type: ignore
438438

439439
for channel in cupy.array(img):
440440
zoom_channel = self._zoom(

0 commit comments

Comments
 (0)