Removed unused mask_params return value (#1626)

Author: clessig

src/weathergen/datasets/multi_stream_data_sampler.py

@@ -379,7 +379,7 @@ def _build_stream_data_input(
                 stream_data.source_is_spoof = rdata.is_spoof
 
                 # preprocess data for model input
-                (source_cells, source_cells_lens, mask_state) = self.tokenizer.get_source(
+                (source_cells, source_cells_lens) = self.tokenizer.get_source(
                     stream_info,
                     rdata,
                     token_data,

src/weathergen/datasets/tokenizer_masking.py

@@ -126,12 +126,7 @@ def get_source(
         if is_diagnostic or rdata.data.shape[1] == 0 or len(rdata.data) < 2:
             source_tokens_cells = [torch.tensor([])]
             source_tokens_lens = torch.zeros([self.num_healpix_cells_source], dtype=torch.int32)
-            mask_state = {
-                "strategy": self.masker.current_strategy,
-                "mask_tokens": None,
-                "mask_channels": None,
-            }
-            return (source_tokens_cells, source_tokens_lens, mask_state)
+            return (source_tokens_cells, source_tokens_lens)
 
         # create tokenization index
         (idxs_cells, idxs_cells_lens) = idxs_cells_data
@@ -154,14 +149,7 @@ def get_source(
             encode_times_source,
         )
 
-        # capture per-view mask state to later produce consistent targets
-        mask_state = {
-            "strategy": None,  # self.masker.current_strategy,
-            "mask_tokens": mask_tokens,
-            "mask_channels": mask_channels,
-        }
-
-        return (source_tokens_cells, source_tokens_lens, mask_state)
+        return (source_tokens_cells, source_tokens_lens)
 
     # batchify_target_for_view now unified into batchify_target via optional mask_state