Skip to content

Commit 1e6e1c8

Browse files
committed
fix bad default for in_channels
Signed-off-by: Carlos Gomes <[email protected]>
1 parent c7d9fec commit 1e6e1c8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/terratorch/models/prithvi_model_factory.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def build_model(
3232
backbone: str | nn.Module,
3333
decoder: str | nn.Module,
3434
bands: list[HLSBands | int],
35-
in_channels: int = int | None, # this should be removed, can be derived from bands. But it is a breaking change
35+
in_channels: int | None = None, # this should be removed, can be derived from bands. But it is a breaking change
3636
num_classes: int | None = None,
3737
pretrained: bool = True, # noqa: FBT001, FBT002
3838
num_frames: int = 1,

0 commit comments

Comments
 (0)