Function _row

experiments/tests/test_knowledge_entity_baseline.py:16–39

(date, paper_id, title, abstract, *, selected=False, oracle_label="irrelevant", **extra)

def _row(date, paper_id, title, abstract, *, selected=False, oracle_label="irrelevant", **extra):
    row = {
        "date": date,
        "episode_id": f"user_role1::{date}",
        "user_id": "user_role1",
        "role_name": "role1",
        "paper_id": paper_id,
        "title": title,
        "abstract": abstract,
        "authors": [],
        "source": "arxiv",
        "url": f"https://example.test/{paper_id}",
        "shown": False,
        "selected": selected,
        "pool_rank": None,
        "system_rank": None,
        "system_score": 0.0,
        "system_label": "edge_relevant",
        "oracle_score": 0.0,
        "oracle_label": oracle_label,
        "select_probability": 0.0,
    }
    row.update(extra)
    return row


def _write_clean_user(input_dir: Path, description="alpha graph agents and benchmark entities"):