Derive the Analysis Age at Randomization

1library(yamaa)23adsl <- yamaa_domain("spec.yaml")$output4adsl

1import yamaa23adsl = yamaa.yamaa_domain("spec.yaml").output4adsl

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.67from datetime import date8from pathlib import Path910import polars as pl111213def is_leap(year: int) -> bool:14    return year % 4 == 0 and (year % 100 != 0 or year % 400 == 0)151617def whole_years(start: date | None, end: date | None) -> int | None:18    """Yearly anniversaries of start on or before end; negated when reversed."""19    if start is None or end is None:20        return None21    if end < start:22        back = whole_years(end, start)23        return None if back is None else -back24    years = end.year - start.year25    month, day = start.month, start.day26    # A February 29 birthday falls on February 28 in common years.27    if month == 2 and day == 29 and not is_leap(end.year):28        day = 2829    if (end.month, end.day) < (month, day):30        years -= 131    return years323334dm = pl.read_csv("/app/input/dm.csv", infer_schema=False).with_columns(35    pl.col("BRTHDT").str.to_date(strict=False),36    pl.col("RANDDT").str.to_date(strict=False),37)3839adsl = (40    dm.with_columns(41        AAGE=pl.struct(["BRTHDT", "RANDDT"]).map_elements(42            lambda r: whole_years(r["BRTHDT"], r["RANDDT"]),43            return_dtype=pl.Int64,44        ),45        AAGEU=pl.lit("YEARS"),46    )47    .select("STUDYID", "USUBJID", "BRTHDT", "RANDDT", "AAGE", "AAGEU")48)4950Path("/app/output").mkdir(exist_ok=True)51adsl.write_csv("/app/output/adsl.csv")

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;