Skip to content

Commit 2794dec

Browse files
committed
Remove mutable default argument
1 parent 93c9354 commit 2794dec

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

monai/networks/blocks/patchembedding.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
from __future__ import annotations
1313

1414
from collections.abc import Sequence
15+
from typing import Optional
1516

1617
import numpy as np
1718
import torch
@@ -53,7 +54,7 @@ def __init__(
5354
pos_embed_type: str = "learnable",
5455
dropout_rate: float = 0.0,
5556
spatial_dims: int = 3,
56-
pos_embed_kwargs: dict = {},
57+
pos_embed_kwargs: Optional[dict] = None,
5758
) -> None:
5859
"""
5960
Args:
@@ -108,6 +109,8 @@ def __init__(
108109
self.position_embeddings = nn.Parameter(torch.zeros(1, self.n_patches, hidden_size))
109110
self.dropout = nn.Dropout(dropout_rate)
110111

112+
pos_embed_kwargs = {} if pos_embed_kwargs is None else pos_embed_kwargs
113+
111114
if self.pos_embed_type == "none":
112115
pass
113116
elif self.pos_embed_type == "learnable":

0 commit comments

Comments
 (0)