Override squeeze methods to return IDataset type for IDataset interface