def _call(self, ds): # Function overrided to let the results have # some dataset attributes res = LinearCSVMC._call(self, ds) if isinstance(res, Dataset): for k in ds.sa.keys(): res.sa[k] = ds.sa[k] return res else: return Dataset(res, sa=ds.sa)