ENH: return stc when applying stc.transform_data