diff --git a/modulus/datapipes/healpix/data_modules.py b/modulus/datapipes/healpix/data_modules.py index 50f7284ba..4c4b48968 100644 --- 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: