def test_squeeze_behavior(self): split = Split(a=None, b=pd.DataFrame(), c=pd.Series(), d=[], e=tuple(), f={}) split.squeeze() self.assertEqual(split, {})
def apply_projection(self, dataset_split: Split) -> Split: """ Identity return """ return dataset_split.squeeze()