app/models/graph/regimen_embedder.py

"""Regimen embedding helper."""

from __future__ import annotations

import numpy as np

from app.models.graph.hetero_encoder import encode_regimen


def regimen_embedding(drugs: list[str]) -> list[float]:
    return encode_regimen(drugs).tolist()