Retain the Latest Earlier Character Result

1library(yamaa)23advs <- yamaa_domain("spec.yaml")$output4advs

1import yamaa23advs = yamaa.yamaa_domain("spec.yaml").output4advs

1# Reference solution for the yamaa benchmark adam-advs-prior-result (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(14    VSSEQ = col_integer(), AVISITN = col_integer(),15    .default = col_character()16  )17)1819# Within a series, rows are ordered by visit number with missing numbers20# last, keeping the collected order for ties. The look-back skips blank21# results but never crosses into another series.22ordered <- vs |>23  arrange(STUDYID, USUBJID, SERIES, is.na(AVISITN), AVISITN)2425# Forward-fill the ordered results, then lag by one so the row's own26# value never counts.27prev <- ordered |>28  group_by(STUDYID, USUBJID, SERIES) |>29  fill(AVALC, .direction = "down") |>30  mutate(PREVAVALC = lag(AVALC)) |>31  ungroup() |>32  select(STUDYID, USUBJID, VSSEQ, PREVAVALC)3334advs <- vs |>35  left_join(prev, by = c("STUDYID", "USUBJID", "VSSEQ")) |>36  arrange(USUBJID, VSSEQ) |>37  select(STUDYID, USUBJID, VSSEQ, SERIES, AVISITN, AVALC, PREVAVALC)3839dir.create("/app/output", showWarnings = FALSE)40write_csv(advs, "/app/output/advs.csv", na = "")

1# Reference solution for the yamaa benchmark adam-advs-prior-result (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.67from pathlib import Path89import polars as pl1011vs = pl.read_csv("/app/input/vs.csv", infer_schema=False).with_columns(12    pl.col("VSSEQ").cast(pl.Int64, strict=False),13    pl.col("AVISITN").cast(pl.Int64, strict=False),14)1516# Rows with no series value share one series within each subject; order17# within a series by visit number with missing numbers last, keeping the18# collected order (input row order) for ties.19ordered = vs.sort(20    ["STUDYID", "USUBJID", "SERIES", "AVISITN"],21    nulls_last=True,22    maintain_order=True,23)2425# The closest earlier result with a value in the same subject and series.26prev_by_key: dict[tuple, str | None] = {}27last_seen: dict[tuple, str | None] = {}28for record in ordered.iter_rows(named=True):29    key = (record["STUDYID"], record["USUBJID"], record["SERIES"])30    prev_by_key[(record["STUDYID"], record["USUBJID"], record["VSSEQ"])] = last_seen.get(key)31    if record["AVALC"] is not None:32        last_seen[key] = record["AVALC"]3334advs = (35    vs.with_columns(36        PREVAVALC=pl.struct(["STUDYID", "USUBJID", "VSSEQ"]).map_elements(37            lambda s: prev_by_key.get((s["STUDYID"], s["USUBJID"], s["VSSEQ"])),38            return_dtype=pl.String,39        )40    )41    .sort(["USUBJID", "VSSEQ"])42    .select("STUDYID", "USUBJID", "VSSEQ", "SERIES", "AVISITN", "AVALC", "PREVAVALC")43)4445Path("/app/output").mkdir(parents=True, exist_ok=True)46advs.write_csv("/app/output/advs.csv")

1/* opensas reference solution for the yamaa benchmark adam-advs-prior-result.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 STUDYID USUBJID SERIES AVALC $1024;7  infile "/app/input/vs.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;8  input STUDYID : $char1024. USUBJID : $char1024. VSSEQ SERIES : $char1024. AVISITN AVALC : $char1024.;9run;1011data ordered; set vs; ordinal=_n_; novisit=missing(AVISITN); run;12proc sort data=ordered; by STUDYID USUBJID SERIES novisit AVISITN ordinal; run;13data result;14  length PREVAVALC carried $1024;15  retain carried;16  set ordered; by STUDYID USUBJID SERIES;17  if first.SERIES then carried='';18  PREVAVALC=carried;19  if not missing(AVALC) then carried=AVALC;20run;2122proc sql;23  create table final as select STUDYID, USUBJID, VSSEQ, SERIES, AVISITN, AVALC, PREVAVALC from result;24quit;25proc export data=final outfile="/app/output/advs.csv" dbms=csv replace; run;