Derive Exposure Duration in Weeks and Months

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-duration (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(lubridate, warn.conflicts = FALSE)9library(readr)1011dm <- read_csv(12  "/app/input/dm.csv",13  col_types = cols(STDT = col_date(), ENDT = col_date(), .default = col_character())14)1516add_months <- function(day, n) {17  total <- as.integer(format(day, "%m")) - 1L + n18  year <- as.integer(format(day, "%Y")) + total %/% 12L19  month <- total %% 12L + 1L20  last <- days_in_month(make_date(year, month, 1L))21  pmin_day <- min(as.integer(format(day, "%d")), last)22  make_date(year, month, pmin_day)23}2425# Monthly anniversaries of the start on or before the end, keeping the26# start day or the month's last day when the month is too short; negated27# with the dates exchanged when the end falls before the start.28whole_months <- function(start, end) {29  if (is.na(start) || is.na(end)) {30    return(NA_integer_)31  }32  if (end < start) {33    return(-whole_months(end, start))34  }35  months <- (as.integer(format(end, "%Y")) - as.integer(format(start, "%Y"))) * 12L +36    (as.integer(format(end, "%m")) - as.integer(format(start, "%m")))37  while (months > 0L && add_months(start, months) > end) {38    months <- months - 1L39  }40  while (add_months(start, months + 1L) <= end) {41    months <- months + 1L42  }43  months44}4546# Whole seven-day blocks from start to end, dropping any leftover partial47# week; negated with the dates exchanged when the end falls before the start.48whole_weeks <- function(start, end) {49  if (is.na(start) || is.na(end)) {50    return(NA_integer_)51  }52  if (end < start) {53    return(-whole_weeks(end, start))54  }55  as.integer(trunc(as.numeric(end - start) / 7))56}5758adsl <- dm |>59  rowwise() |>60  mutate(DURW = whole_weeks(STDT, ENDT), DURM = whole_months(STDT, ENDT)) |>61  ungroup() |>62  select(STUDYID, USUBJID, STDT, ENDT, DURW, DURM)6364dir.create("/app/output", showWarnings = FALSE)65write_csv(adsl, "/app/output/adsl.csv", na = "")

1# Reference solution for the yamaa benchmark adam-adsl-duration (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 calendar8from datetime import date9from pathlib import Path1011import polars as pl121314def add_months(day: date, n: int) -> date:15    total = day.month - 1 + n16    year = day.year + total // 1217    month = total % 12 + 118    last = calendar.monthrange(year, month)[1]19    return date(year, month, min(day.day, last))202122def whole_months(start: date | None, end: date | None) -> int | None:23    """Monthly anniversaries of start on or before end; negated when reversed."""24    if start is None or end is None:25        return None26    if end < start:27        back = whole_months(end, start)28        return None if back is None else -back29    months = (end.year - start.year) * 12 + (end.month - start.month)30    while months > 0 and add_months(start, months) > end:31        months -= 132    while add_months(start, months + 1) <= end:33        months += 134    return months353637def whole_weeks(start: date | None, end: date | None) -> int | None:38    """Whole seven-day blocks from start to end; negated when reversed."""39    if start is None or end is None:40        return None41    if end < start:42        back = whole_weeks(end, start)43        return None if back is None else -back44    return (end - start).days // 7454647dm = pl.read_csv("/app/input/dm.csv", infer_schema=False).with_columns(48    pl.col("STDT").str.to_date(strict=False),49    pl.col("ENDT").str.to_date(strict=False),50)5152adsl = (53    dm.with_columns(54        DURW=pl.struct(["STDT", "ENDT"]).map_elements(55            lambda r: whole_weeks(r["STDT"], r["ENDT"]), return_dtype=pl.Int6456        ),57        DURM=pl.struct(["STDT", "ENDT"]).map_elements(58            lambda r: whole_months(r["STDT"], r["ENDT"]), return_dtype=pl.Int6459        ),60    )61    .select("STUDYID", "USUBJID", "STDT", "ENDT", "DURW", "DURM")62)6364Path("/app/output").mkdir(exist_ok=True)65adsl.write_csv("/app/output/adsl.csv")

1/* opensas reference solution for the yamaa benchmark adam-adsl-duration.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 STDT ENDT $1024;7  infile "/app/input/dm.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;8  input STUDYID : $char1024. USUBJID : $char1024. STDT : $char1024. ENDT : $char1024.;9run;1011data result;12  set dm;13  start=input(STDT,yymmdd10.); finish=input(ENDT,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  DURW=sign*floor((hi-lo)/7); DURM=sign*intck('month',lo,hi,'continuous');18  end;19run;2021proc sql;22  create table final as select STUDYID, USUBJID, STDT, ENDT, DURW, DURM from result;23quit;24proc export data=final outfile="/app/output/adsl.csv" dbms=csv replace; run;