We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c7d9fec commit 1e6e1c8Copy full SHA for 1e6e1c8
src/terratorch/models/prithvi_model_factory.py
@@ -32,7 +32,7 @@ def build_model(
32
backbone: str | nn.Module,
33
decoder: str | nn.Module,
34
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
+ in_channels: int | None = None, # this should be removed, can be derived from bands. But it is a breaking change
36
num_classes: int | None = None,
37
pretrained: bool = True, # noqa: FBT001, FBT002
38
num_frames: int = 1,
0 commit comments