Function get_dummy_experience

tests/explorer/proxy_test.py:12–23

(num: int)

def get_dummy_experience(num: int) -> List[Experience]:
    return [
        Experience(
            eid=EID(suffix=uuid.uuid4().hex[:6]),
            tokens=torch.zeros(5),
            prompt_length=2,
            info={
                "model_version": 0,
            },
        )
        for _ in range(num)
    ]


db_path = os.path.join(os.path.dirname(__file__), "test_recorder.db")