Ordered Float Sum

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

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

1# Reference solution for the yamaa benchmark adam-adlb-ordered-sum (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)910lb <- read_csv(11  "/app/input/lb.csv",12  col_types = cols(13    LBSEQ = col_integer(),14    LBSTRESN = col_double(),15    .default = col_character()16  )17)1819# Keep every collected component record; DTYPE has no value on these.20components <- lb |>21  transmute(22    STUDYID, USUBJID,23    PARAMCD = LBTESTCD,24    PARAM = LBTEST,25    AVISIT = VISIT,26    AVAL = LBSTRESN,27    DTYPE = NA_character_28  )2930# One total record per subject and visit: the sum of the collected31# component results in laboratory sequence order. No value, rather than32# zero, when none of them has a collected result.33ordered_sum <- function(x) {34  vals <- x[!is.na(x)]35  if (length(vals) == 0) {36    return(NA_real_)37  }38  total <- vals[[1]]39  if (length(vals) > 1) {40    for (v in vals[-1]) {41      total <- total + v42    }43  }44  total45}4647totals <- lb |>48  arrange(USUBJID, VISIT, LBSEQ) |>49  group_by(STUDYID, USUBJID, VISIT) |>50  summarise(51    AVAL = ordered_sum(LBSTRESN),52    .groups = "drop"53  ) |>54  mutate(55    PARAMCD = "TOTAL",56    PARAM = "Total of Components",57    AVISIT = VISIT,58    DTYPE = "CALCULATION"59  ) |>60  select(STUDYID, USUBJID, PARAMCD, PARAM, AVISIT, AVAL, DTYPE)6162adlb <- bind_rows(components, totals)6364dir.create("/app/output", showWarnings = FALSE)65write_csv(adlb, "/app/output/adlb.csv", na = "")

1# Reference solution for the yamaa benchmark adam-adlb-ordered-sum (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 ADLB from LB: every collected record plus one TOTAL record8per subject and visit, summed in the order the source records were9stored."""1011import csv12import os1314INPUT_DIR = "/app/input"15OUTPUT_DIR = "/app/output"1617COLUMNS = ["STUDYID", "USUBJID", "PARAMCD", "PARAM", "AVISIT", "AVAL", "DTYPE"]181920def main():21    with open(os.path.join(INPUT_DIR, "lb.csv"), newline="") as f:22        source = list(csv.DictReader(f))2324    collected = []  # (group key, output row) in stored order25    totals = {}  # group key -> running total, None until a result is seen26    last_index = {}  # group key -> index in `collected` of its last record2728    for row in source:29        key = (row["STUDYID"], row["USUBJID"], row["VISIT"])30        result = (row["LBSTRESN"] or "").strip()31        totals.setdefault(key, None)32        if result != "":33            value = float(result)34            totals[key] = value if totals[key] is None else totals[key] + value35        collected.append(36            (37                key,38                {39                    "STUDYID": row["STUDYID"],40                    "USUBJID": row["USUBJID"],41                    "PARAMCD": row["LBTESTCD"],42                    "PARAM": row["LBTEST"],43                    "AVISIT": row["VISIT"],44                    "AVAL": result,45                    "DTYPE": "",46                },47            )48        )49        last_index[key] = len(collected) - 15051    final_rows = []52    for i, (key, row) in enumerate(collected):53        final_rows.append(row)54        if last_index[key] == i:55            total = totals[key]56            final_rows.append(57                {58                    "STUDYID": key[0],59                    "USUBJID": key[1],60                    "PARAMCD": "TOTAL",61                    "PARAM": "Total of Components",62                    "AVISIT": key[2],63                    "AVAL": "" if total is None else repr(total),64                    "DTYPE": "CALCULATION",65                }66            )6768    os.makedirs(OUTPUT_DIR, exist_ok=True)69    with open(os.path.join(OUTPUT_DIR, "adlb.csv"), "w", newline="") as f:70        writer = csv.DictWriter(f, fieldnames=COLUMNS)71        writer.writeheader()72        writer.writerows(final_rows)737475if __name__ == "__main__":76    main()

1/* opensas reference solution for the yamaa benchmark adam-adlb-ordered-sum.2   Reads the benchmark inputs and derives the requested output.3   Execute with the existing opensas runtime: opensas /app/output/result.sas. */45data lb;6  length STUDYID USUBJID LBTESTCD LBTEST VISIT $1024;7  infile "/app/input/lb.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;8  input STUDYID : $char1024. USUBJID : $char1024. LBSEQ LBTESTCD : $char1024. LBTEST : $char1024. VISIT : $char1024. LBSTRESN;9run;1011data collected;12  length PARAMCD $16 PARAM $1024 AVISIT $1024 DTYPE $11;13  set lb; PARAMCD=LBTESTCD; PARAM=LBTEST; AVISIT=VISIT; AVAL=LBSTRESN;14run;15proc sort data=collected; by STUDYID USUBJID AVISIT LBSEQ; run;16data totals;17  length PARAMCD $16 PARAM $1024 DTYPE $11;18  set collected; by STUDYID USUBJID AVISIT;19  if first.AVISIT then do; total=0; count=0; end;20  if not missing(AVAL) then do; total+AVAL; count+1; end;21  if last.AVISIT then do;22    PARAMCD='TOTAL'; PARAM='Total of Components'; DTYPE='CALCULATION';23    AVAL=.; if count>0 then AVAL=total;24    output;25  end;26run;27data result; set collected totals; run;2829proc sql;30  create table final as select STUDYID, USUBJID, PARAMCD, PARAM, AVISIT, AVAL, DTYPE from result;31quit;32proc export data=final outfile="/app/output/adlb.csv" dbms=csv replace; run;