def transform_data(data: DataFrame, date: str) -> DataFrame:
"""Transform original dataset.
:param data: Input DataFrame.
:param date: The context date
:return: Transformed DataFrame.
"""
return data.transform(add_ds(date)).dropDuplicates()
