new training unit test

[prishavall]

New Training Unit Test for test sk train dataset

Github Issue Number Here: #1016 What user problem are we solving? Tests if SklearnDatasetCreator can correctly split a default dataset into train and test datasets given the dataset's name What solution does this PR provide? Looks at individual variable counts for batch sizes and tests to see if they are the right lengths Testing Methodology

How did you test your changes and verify that existing functionality is not broken Ran tests and checked if variables matched up Any other considerations

[sweep-ai[bot]]

Apply Sweep Rules to your PR?

• [ ] Apply: Leftover TODOs in the code should be handled.
• [ ] Apply: All new business logic should have corresponding unit tests in the tests/ directory.
• [ ] Apply: Any clearly inefficient or repeated code should be optimized or refactored.