Skip to content

Commit

Permalink
HLSBands also need be considered
Browse files Browse the repository at this point in the history
Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
  • Loading branch information
Joao-L-S-Almeida committed Aug 9, 2024
1 parent 6bd2bde commit 52f0c12
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion terratorch/models/backbones/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ def _estimate_in_chans(model_bands: list[HLSBands] | list[str] | tuple[int, int]

# Conditional to deal with the different possible choices for the bands
# Bands as lists of strings or enum
if all([isinstance(b, str) for b in model_bands]):
if all([isinstance(b, str) or isinstance(b, HLSBands) for b in model_bands]):
in_chans = len(model_bands)
# Bands as intervals limited by integers
elif all([isinstance(b, int) for b in model_bands] or _are_sublists_of_int(model_bands)):
Expand Down

0 comments on commit 52f0c12

Please sign in to comment.