First Rescue Medication

1library(yamaa)23yamaa_spec("spec.yaml")$run()$output

1import yamaa23yamaa.spec("spec.yaml").run().output

1# Reference solution for the yamaa benchmark adam-adsl-rescue-med (R track).2#3# Written from the benchmark's full prompt and its inputs alone, as an agent4# would write it. Harbor's oracle agent runs it to check that the prompt can5# be solved and that the grader scores a correct answer 1.67library(dplyr, warn.conflicts = FALSE)8library(readr)910dm <- read_csv(11  "/app/input/dm.csv",12  col_types = cols(.default = col_character())13)14cm <- read_csv(15  "/app/input/cm.csv",16  col_types = cols(CMSEQ = col_integer(), .default = col_character())17)1819# The first rescue medication by earliest start date, with the smaller20# sequence number breaking a tie. A start holding only year and month21# compares as written, so it falls before any full date in the same22# month; a record with no start date still counts.23first <- cm |>24  filter(CMCAT == "RESCUE MEDICATION") |>25  mutate(sortkey = coalesce(CMSTDTC, "")) |>26  arrange(sortkey, CMSEQ) |>27  distinct(STUDYID, USUBJID, .keep_all = TRUE) |>28  select(STUDYID, USUBJID, RESCTRT = CMTRT)2930adsl <- dm |>31  left_join(first, by = c("STUDYID", "USUBJID")) |>32  select(STUDYID, USUBJID, RESCTRT) |>33  arrange(USUBJID)3435dir.create("/app/output", showWarnings = FALSE)36write_csv(adsl, "/app/output/adsl.csv", na = "")

1# Reference solution for the yamaa benchmark adam-adsl-rescue-med (Python track).2#3# Written from the benchmark's full prompt and its inputs alone, as an agent4# would write it. Harbor's oracle agent runs it to check that the prompt can5# be solved and that the grader scores a correct answer 1.67import csv8from pathlib import Path910INPUT = Path("/app/input")11OUTPUT = Path("/app/output")121314def read_rows(name):15    with open(INPUT / name, newline="", encoding="utf-8") as f:16        return list(csv.DictReader(f))171819def main():20    dm = read_rows("dm.csv")21    cm = read_rows("cm.csv")2223    rescue = {}24    for row in cm:25        if (row.get("CMCAT") or "") != "RESCUE MEDICATION":26            continue27        key = (row.get("STUDYID") or "", row.get("USUBJID") or "")28        # Compare the start date as written: a year-month value such as29        # "2025-03" sorts before any full date in the same month under30        # plain string ordering. A missing date sorts as the empty string.31        date_key = row.get("CMSTDTC") or ""32        try:33            seq_key = int(row.get("CMSEQ") or "")34        except ValueError:35            try:36                seq_key = float(row.get("CMSEQ") or "")37            except ValueError:38                seq_key = 039        sort_key = (date_key, seq_key)40        if key not in rescue or sort_key < rescue[key][0]:41            rescue[key] = (sort_key, row.get("CMTRT") or "")4243    OUTPUT.mkdir(parents=True, exist_ok=True)44    with open(OUTPUT / "adsl.csv", "w", newline="", encoding="utf-8") as f:45        writer = csv.writer(f)46        writer.writerow(["STUDYID", "USUBJID", "RESCTRT"])47        for row in dm:48            studyid = row.get("STUDYID") or ""49            usubjid = row.get("USUBJID") or ""50            entry = rescue.get((studyid, usubjid))51            resctrt = entry[1] if entry else ""52            writer.writerow([studyid, usubjid, resctrt])535455if __name__ == "__main__":56    main()

1/* opensas reference solution for the yamaa benchmark adam-adsl-rescue-med.2   Reads the benchmark inputs and derives the requested output.3   Execute with the existing opensas runtime: opensas /app/output/result.sas. */45data cm;6  length STUDYID USUBJID CMTRT CMCAT CMSTDTC $1024;7  infile "/app/input/cm.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;8  input STUDYID : $char1024. USUBJID : $char1024. CMSEQ CMTRT : $char1024. CMCAT : $char1024. CMSTDTC : $char1024.;9run;1011data dm;12  length STUDYID USUBJID $1024;13  infile "/app/input/dm.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;14  input STUDYID : $char1024. USUBJID : $char1024.;15run;1617data rescue; set cm; if CMCAT='RESCUE MEDICATION'; nodate=missing(CMSTDTC); run;18proc sort data=rescue; by STUDYID USUBJID nodate CMSTDTC CMSEQ; run;19data firstcm; set rescue; by STUDYID USUBJID; if first.USUBJID; run;20proc sql; create table result as select a.*,b.CMTRT as RESCTRT from dm as a left join firstcm as b on a.STUDYID=b.STUDYID and a.USUBJID=b.USUBJID; quit;2122proc export data=result(keep=STUDYID USUBJID RESCTRT) outfile="/app/output/adsl.csv" dbms=csv replace; run;