Key 'partition' is not expected

```shell
usage: terratorch [-h] [--defaults CONFIG]
[--defaults.trainer_args TRAINER_ARGS]
[--defaults.terratorch_task TERRATORCH_TASK]
[--optimization_space OPTIMIZATION_SPACE]
[--experiment_name EXPERIMENT_NAME] [--run_name RUN_NAME]
[--save_models {true,false}] [--storage_uri STORAGE_URI]
[--ray_storage_path RAY_STORAGE_PATH] [--n_trials N_TRIALS]
[--run_repetitions RUN_REPETITIONS] [--tasks TASKS]
[--parent_run_id PARENT_RUN_ID] [--output_path OUTPUT_PATH]
[--logger LOGGER] [--config CONFIG]
[--print_config [=flags]] [--hpo] [--repeat]
error: Parser key "datamodule":
Problem with given class_path 'terratorch.datamodules.GenericNonGeoPixelwiseRegressionDataModule':
Validation failed: Key 'partition' is not expected
```

note that this is a terratorch module:`terratorch.datamodules.GenericNonGeoPixelwiseRegressionDataModule`, if I'm not mistaken, this [one]( https://github.com/IBM/terratorch/blob/main/terratorch/datamodules/generic_pixel_wise_data_module.py). My guess is that there is an inconsistency between the config file and the terratorch version