Fix for dlwp_healpix dataloader not including coupled variables (#702)

Co-authored-by: David Pruitt <dpruitt@nvidia.com>
NVIDIA · Nov 18, 2024 · bedf90b · bedf90b
1 parent 26efe42
commit bedf90b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modulus/datapipes/healpix/data_modules.py b/modulus/datapipes/healpix/data_modules.py
@@ -989,7 +989,7 @@ def setup(self) -> None:
                 )
 
                 dataset = dataset.sel(
-                    channel_in=self.input_variables,
+                    channel_in=self.input_variables + coupled_variables,
                     channel_out=self.output_variables,
                 )
             else: