Issue integrating new dataset and custom TrainTestSplitter

[igoramaralms]

Hi!

I'm currently testing the framework with the CALCE dataset before applying it to my own battery dataset.

I created a custom CALCETrainTestSplitter class, added a new YAML configuration, and made the necessary updates to the data loading and preprocessing logic.

However, when trying to run the pipeline, I get the following error:

raise KeyError(f'{name} is not registered to {self.name}!')
KeyError: 'CALCETrainTestSplitter is not registered to Train Test Splitters!'

It seems that the framework doesn't recognize the custom splitter, even though the class is implemented and referenced in the config.

Is there an extra step needed to properly register custom TrainTestSplitter classes so they can be used with new datasets?

Thanks in advance for the support!

[fingertap]

You should:

1. decorate the class with the register method. See other splitters for example.
2. import this class such that the registry actually runs register on your class, so that the pipeline can build your class instance from string (defined in the config yaml).