Derive Exposure Duration in Weeks and Months

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

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

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 csv8import calendar9from datetime import date1011IN_DIR = "/app/input"12OUT_DIR = "/app/output"131415def parse_date(text):16    text = (text or "").strip()17    if not text:18        return None19    return date.fromisoformat(text)202122def add_months(d, months):23    """Date `months` months after d, on the same day of month, or on the24    last day of the target month when it has no such day."""25    total = (d.year * 12 + (d.month - 1)) + months26    year, month0 = divmod(total, 12)27    month = month0 + 128    day = min(d.day, calendar.monthrange(year, month)[1])29    return date(year, month, day)303132def durations(start, end):33    """(DURW, DURM) from start to end; both blank if either is missing."""34    if start is None or end is None:35        return "", ""36    sign = 137    lo, hi = start, end38    if hi < lo:39        sign = -140        lo, hi = hi, lo41    durw = (hi - lo).days // 742    durm = 043    while add_months(lo, durm + 1) <= hi:44        durm += 145    return sign * durw, sign * durm464748def main():49    with open(f"{IN_DIR}/dm.csv", newline="") as f:50        rows = list(csv.DictReader(f))5152    out_rows = []53    for row in rows:54        stdt = parse_date(row.get("STDT"))55        endt = parse_date(row.get("ENDT"))56        durw, durm = durations(stdt, endt)57        out_rows.append(58            {59                "STUDYID": row.get("STUDYID", ""),60                "USUBJID": row.get("USUBJID", ""),61                "STDT": row.get("STDT", ""),62                "ENDT": row.get("ENDT", ""),63                "DURW": durw,64                "DURM": durm,65            }66        )6768    fieldnames = ["STUDYID", "USUBJID", "STDT", "ENDT", "DURW", "DURM"]69    with open(f"{OUT_DIR}/adsl.csv", "w", newline="") as f:70        writer = csv.DictWriter(f, fieldnames=fieldnames)71        writer.writeheader()72        writer.writerows(out_rows)737475if __name__ == "__main__":76    main()

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 export data=result(keep=STUDYID USUBJID STDT ENDT DURW DURM) outfile="/app/output/adsl.csv" dbms=csv replace; run;