CPM-9G-8B/quick_start_clean/transform_data.py

import random


def rand(n: int, r: random.Random):
    return int(r.random() * n)


def transform(data, num_sample: int, r: random.Random):
    return {"input": data["input"], "output": data["output"]}