Skip to content

Commit 42e03c6

Browse files
committed
remove unneded method
1 parent 6b089c9 commit 42e03c6

1 file changed

Lines changed: 0 additions & 10 deletions

File tree

olmoearth_pretrain/train/masking.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1001,8 +1001,6 @@ def _select_bandsets_for_sample(
10011001
"""Select encoded and decoded bandsets for a single sample based on modality count."""
10021002
if len(present) == 1:
10031003
return self._select_bandsets_single_modality(present)
1004-
elif len(present) == 2:
1005-
return self._select_bandsets_two_modalities(present)
10061004
else:
10071005
return self._select_bandsets_multiple_modalities(present)
10081006

@@ -1014,14 +1012,6 @@ def _select_bandsets_single_modality(
10141012
decoded_bandset_idxs: set = set()
10151013
return encoded_bandset_idxs, decoded_bandset_idxs
10161014

1017-
def _select_bandsets_two_modalities(
1018-
self, present: list[tuple[str, int]]
1019-
) -> tuple[set[tuple[str, int]], set[tuple[str, int]]]:
1020-
"""Handle two modality case: encode first, decode second."""
1021-
encoded_bandset_idxs = set([present[0]])
1022-
decoded_bandset_idxs = set([present[1]])
1023-
return encoded_bandset_idxs, decoded_bandset_idxs
1024-
10251015
def _select_bandsets_multiple_modalities(
10261016
self, present: list[tuple[str, int]]
10271017
) -> tuple[set[tuple[str, int]], set[tuple[str, int]]]:

0 commit comments

Comments
 (0)