Carry the last observed value across missing visits

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

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

1# Reference solution for the yamaa benchmark adam-advs-locf (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)9library(tidyr)1011vs <- read_csv(12  "/app/input/vs.csv",13  col_types = cols(AVISITN = col_integer(), AVALCOL = col_double(), .default = col_character())14)1516# The collected value, otherwise the closest earlier visit with one, in17# visit-number order; a gap before the first stays missing.18advs <- vs |>19  arrange(USUBJID, PARAMCD, AVISITN) |>20  group_by(USUBJID, PARAMCD) |>21  fill(AVALCOL, .direction = "down") |>22  mutate(AVAL = AVALCOL) |>23  ungroup() |>24  select(USUBJID, PARAMCD, AVISITN, AVAL)2526dir.create("/app/output", showWarnings = FALSE)27write_csv(advs, "/app/output/advs.csv", na = "")

1# Reference solution for the yamaa benchmark adam-advs-locf (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.67"""Create the ADVS dataset from the VS dataset.89One record per subject per parameter per analysis visit. AVAL is the10collected value when the record has one; otherwise it is the value from11the closest earlier visit, in visit-number order, that has one, for the12same subject and parameter (last observation carried forward). A gap13before the first collected value has no value. A collected value of14zero is carried like any other value.15"""1617import csv1819INPUT_PATH = "/app/input/vs.csv"20OUTPUT_PATH = "/app/output/advs.csv"212223def main() -> None:24    with open(INPUT_PATH, newline="") as f:25        rows = list(csv.DictReader(f))2627    # Walk the records in subject / parameter / visit-number order so28    # each record can see the closest earlier collected value of its29    # group, then map the results back to the original record order.30    order = sorted(31        range(len(rows)),32        key=lambda i: (33            rows[i]["USUBJID"],34            rows[i]["PARAMCD"],35            int(rows[i]["AVISITN"]),36        ),37    )3839    last_value = {}40    aval = [""] * len(rows)41    for i in order:42        row = rows[i]43        key = (row["USUBJID"], row["PARAMCD"])44        collected = row["AVALCOL"]45        if collected != "":46            last_value[key] = collected47            aval[i] = collected48        else:49            aval[i] = last_value.get(key, "")5051    with open(OUTPUT_PATH, "w", newline="") as f:52        writer = csv.writer(f)53        writer.writerow(["USUBJID", "PARAMCD", "AVISITN", "AVAL"])54        for i, row in enumerate(rows):55            writer.writerow(56                [row["USUBJID"], row["PARAMCD"], row["AVISITN"], aval[i]]57            )585960if __name__ == "__main__":61    main()

1/* opensas reference solution for the yamaa benchmark adam-advs-locf.2   Reads the benchmark inputs and derives the requested output.3   Execute with the existing opensas runtime: opensas /app/output/result.sas. */45data vs;6  length USUBJID PARAMCD $1024;7  infile "/app/input/vs.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;8  input USUBJID : $char1024. PARAMCD : $char1024. AVISITN AVALCOL;9run;1011proc sort data=vs out=ordered; by USUBJID PARAMCD AVISITN; run;12data result;13  set ordered; by USUBJID PARAMCD;14  retain carried;15  if first.PARAMCD then carried=.;16  if not missing(AVALCOL) then carried=AVALCOL;17  AVAL=carried;18run;1920proc export data=result(keep=USUBJID PARAMCD AVISITN AVAL) outfile="/app/output/advs.csv" dbms=csv replace; run;