I thought about giving the dataset as both input and output to have the keys in the load() method and the save() but Kedro complains that it cannot be