Derive the Analysis Age at Randomization

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

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

1# Reference solution for the yamaa benchmark adam-adsl-analysis-age (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(BRTHDT = col_date(), RANDDT = col_date(), .default = col_character())13)1415is_leap <- function(year) {16  year %% 4 == 0 & (year %% 100 != 0 | year %% 400 == 0)17}1819# Yearly anniversaries of the birth date on or before randomization; a20# February 29 birthday falls on February 28 in common years, and a date21# before birth gives the negated count with the dates exchanged.22whole_years <- function(start, end) {23  if (is.na(start) || is.na(end)) {24    return(NA_integer_)25  }26  if (end < start) {27    return(-whole_years(end, start))28  }29  years <- as.integer(format(end, "%Y")) - as.integer(format(start, "%Y"))30  month <- as.integer(format(start, "%m"))31  day <- as.integer(format(start, "%d"))32  if (month == 2L && day == 29L && !is_leap(as.integer(format(end, "%Y")))) {33    day <- 28L34  }35  end_month <- as.integer(format(end, "%m"))36  end_day <- as.integer(format(end, "%d"))37  if (end_month < month || (end_month == month && end_day < day)) {38    years <- years - 1L39  }40  years41}4243adsl <- dm |>44  rowwise() |>45  mutate(AAGE = whole_years(BRTHDT, RANDDT), AAGEU = "YEARS") |>46  ungroup() |>47  select(STUDYID, USUBJID, BRTHDT, RANDDT, AAGE, AAGEU)4849dir.create("/app/output", showWarnings = FALSE)50write_csv(adsl, "/app/output/adsl.csv", na = "")

1# Reference solution for the yamaa benchmark adam-adsl-analysis-age (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 csv8import os9from datetime import date101112def parse_date(text):13    text = (text or "").strip()14    if not text:15        return None16    year, month, day = text.split("-")17    return date(int(year), int(month), int(day))181920def anniversary_in_year(birth, year):21    """Yearly anniversary of `birth` within `year`.2223    A February 29 birthday falls on February 28 in a common year.24    """25    month, day = birth.month, birth.day26    if month == 2 and day == 29:27        try:28            return date(year, 2, 29)29        except ValueError:30            return date(year, 2, 28)31    return date(year, month, day)323334def count_anniversaries(start, end):35    """Count of yearly anniversaries of `start` falling on or before `end`.3637    Assumes end >= start.38    """39    years = end.year - start.year40    if anniversary_in_year(start, end.year) > end:41        years -= 142    return years434445def analysis_age(brthdt, randdt):46    if brthdt is None or randdt is None:47        return ""48    if randdt >= brthdt:49        return count_anniversaries(brthdt, randdt)50    return -count_anniversaries(randdt, brthdt)515253def main():54    with open("/app/input/dm.csv", newline="") as f:55        rows = list(csv.DictReader(f))5657    out_rows = []58    for row in rows:59        brthdt = parse_date(row["BRTHDT"])60        randdt = parse_date(row["RANDDT"])61        out_rows.append(62            {63                "STUDYID": row["STUDYID"],64                "USUBJID": row["USUBJID"],65                "BRTHDT": row["BRTHDT"],66                "RANDDT": row["RANDDT"],67                "AAGE": analysis_age(brthdt, randdt),68                "AAGEU": "YEARS",69            }70        )7172    os.makedirs("/app/output", exist_ok=True)73    with open("/app/output/adsl.csv", "w", newline="") as f:74        writer = csv.DictWriter(75            f,76            fieldnames=["STUDYID", "USUBJID", "BRTHDT", "RANDDT", "AAGE", "AAGEU"],77        )78        writer.writeheader()79        writer.writerows(out_rows)808182if __name__ == "__main__":83    main()

1/* opensas reference solution for the yamaa benchmark adam-adsl-analysis-age.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 BRTHDT RANDDT $1024;7  infile "/app/input/dm.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;8  input STUDYID : $char1024. USUBJID : $char1024. BRTHDT : $char1024. RANDDT : $char1024.;9run;1011data result;12  set dm;13  start=input(BRTHDT,yymmdd10.); finish=input(RANDDT,yymmdd10.);14  if not missing(start) and not missing(finish) then do;15    lo=min(start,finish); hi=max(start,finish); sign=1;16    if finish<start then sign=-1;17  AAGE=sign*intck('year',lo,hi,'continuous');18  end;19  length AAGEU $5; AAGEU='YEARS';20run;2122proc sql;23  create table final as select STUDYID, USUBJID, BRTHDT, RANDDT, AAGE, AAGEU from result;24quit;25proc export data=final outfile="/app/output/adsl.csv" dbms=csv replace; run;