Skip to content

Commit c83f247

Browse files
committed
merge again
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 0f872ef commit c83f247

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/compressed_tensors/utils/match.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636

3737
def match_named_modules(
3838
model: torch.nn.Module,
39-
targets: Iterable[str] = tuple(),
39+
targets: Iterable[str],
4040
ignore: Iterable[str] = tuple(),
4141
warn_on_fail: bool = False,
4242
) -> Generator[Tuple[str, torch.nn.Module]]:
@@ -69,7 +69,7 @@ def match_named_modules(
6969
def match_named_parameters(
7070
model: torch.nn.Module,
7171
targets: Iterable[str],
72-
ignore: Iterable[str],
72+
ignore: Iterable[str] = tuple(),
7373
warn_on_fail: bool = False,
7474
) -> Generator[Tuple[str, torch.nn.Module, torch.nn.Parameter]]:
7575
"""
@@ -103,7 +103,7 @@ def match_named_parameters(
103103
def match_modules_set(
104104
model: torch.nn.Module,
105105
targets: Iterable[str],
106-
ignore: Iterable[str],
106+
ignore: Iterable[str] = tuple(),
107107
) -> Generator[Iterable[torch.nn.Module]]:
108108
"""
109109
Yields modules grouped with the same order and size as `targets`.

0 commit comments

Comments
 (0)