polyguard-rl/app/models/graph/pairwise_ddi_head.py

"""Pairwise DDI head."""

from __future__ import annotations

from app.knowledge.ddi_knowledge import is_contraindicated_pair


def score_pair(drug_a: str, drug_b: str) -> float:
    return 0.95 if is_contraindicated_pair(drug_a, drug_b) else 0.15