Pool Subjects into Age Groups

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

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

1# Reference solution for the yamaa benchmark adam-adsl-age-group (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(AGE = col_double(), .default = col_character())13)1415adsl <- dm |>16  mutate(17    AGEGR1 = case_when(18      is.na(AGE) ~ "Missing",19      AGE < 18 ~ "<18",20      AGE <= 64 ~ "18-64",21      .default = ">64"22    ),23    AGEGR1N = case_when(24      AGEGR1 == "<18" ~ 1L,25      AGEGR1 == "18-64" ~ 2L,26      AGEGR1 == ">64" ~ 3L27    )28  ) |>29  select(STUDYID, USUBJID, AGE, AGEU, AGEGR1, AGEGR1N)3031dir.create("/app/output", showWarnings = FALSE)32write_csv(adsl, "/app/output/adsl.csv", na = "")

1# Reference solution for the yamaa benchmark adam-adsl-age-group (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 csv89INPUT_CSV = "/app/input/dm.csv"10OUTPUT_CSV = "/app/output/adsl.csv"111213def age_group(age_text):14    """Return (AGEGR1, AGEGR1N) for a raw AGE cell."""15    if age_text is None or age_text.strip() == "":16        return "Missing", ""17    age = float(age_text)18    if age < 18:19        return "<18", "1"20    if age <= 64:21        return "18-64", "2"22    return ">64", "3"232425def main():26    with open(INPUT_CSV, newline="") as f:27        rows = list(csv.DictReader(f))2829    out_rows = []30    for row in rows:31        age = row["AGE"]32        agegr1, agegr1n = age_group(age)33        out_rows.append(34            {35                "STUDYID": row["STUDYID"],36                "USUBJID": row["USUBJID"],37                "AGE": age,38                "AGEU": row["AGEU"],39                "AGEGR1": agegr1,40                "AGEGR1N": agegr1n,41            }42        )4344    fieldnames = ["STUDYID", "USUBJID", "AGE", "AGEU", "AGEGR1", "AGEGR1N"]45    with open(OUTPUT_CSV, "w", newline="") as f:46        writer = csv.DictWriter(f, fieldnames=fieldnames)47        writer.writeheader()48        writer.writerows(out_rows)495051if __name__ == "__main__":52    main()

1/* opensas reference solution for the yamaa benchmark adam-adsl-age-group.2   Reads the benchmark inputs and derives the requested output.3   Execute with the existing opensas runtime: opensas /app/output/result.sas. */45data dm;6  length STUDYID USUBJID AGEU $1024;7  infile "/app/input/dm.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;8  input STUDYID : $char1024. USUBJID : $char1024. AGE AGEU : $char1024.;9run;1011data result;12  length AGEGR1 $7;13  set dm;14  if missing(AGE) then do; AGEGR1='Missing'; AGEGR1N=.; end;15  else if AGE < 18 then do; AGEGR1='<18'; AGEGR1N=1; end;16  else if AGE <= 64 then do; AGEGR1='18-64'; AGEGR1N=2; end;17  else do; AGEGR1='>64'; AGEGR1N=3; end;18run;1920proc sql;21  create table final as select STUDYID, USUBJID, AGE, AGEU, AGEGR1, AGEGR1N from result;22quit;23proc export data=final outfile="/app/output/adsl.csv" dbms=csv replace; run;