Dose Reduction Flag

1library(yamaa)23adex <- yamaa_domain("spec.yaml")$output4adex

1import yamaa23adex = yamaa.yamaa_domain("spec.yaml").output4adex

1# Reference solution for the yamaa benchmark adam-adex-dose-reduction (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)910ex <- read_csv(11  "/app/input/ex.csv",12  col_types = cols(EXSEQ = col_integer(), .default = col_character())13)1415adex <- ex |>16  mutate(17    EXSTDTM = if_else(18      !is.na(EXSTDTM) & nchar(EXSTDTM) == 16L, paste0(EXSTDTM, ":00"), EXSTDTM19    ),20    EXDOSE = suppressWarnings(as.numeric(EXDOSE))21  ) |>22  # Chronological treatment-start order within each subject, breaking23  # timestamp ties by sequence number.24  arrange(USUBJID, EXSTDTM, EXSEQ) |>25  group_by(USUBJID) |>26  mutate(27    PREVDOSE = lag(EXDOSE),28    # Y when the current dose is lower than the previous one; no value for29    # the first record and whenever either dose is zero or missing. The30    # previous dose is always the administration just before.31    DOSREDFL = if_else(32      !is.na(EXDOSE) & EXDOSE != 0 & !is.na(PREVDOSE) & PREVDOSE != 0 &33        EXDOSE < PREVDOSE,34      "Y",35      NA_character_36    )37  ) |>38  ungroup() |>39  select(STUDYID, USUBJID, EXSEQ, EXSTDTM, EXDOSE, DOSREDFL)4041dir.create("/app/output", showWarnings = FALSE)42write_csv(adex, "/app/output/adex.csv", na = "")

1# Reference solution for the yamaa benchmark adam-adex-dose-reduction (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 pathlib import Path89import polars as pl1011ex = pl.read_csv("/app/input/ex.csv", infer_schema=False).with_columns(12    pl.col("EXSEQ").cast(pl.Int64)13)1415adex = (16    ex.with_columns(17        EXSTDTM=pl.when(pl.col("EXSTDTM").str.len_chars() == 16)18        .then(pl.col("EXSTDTM") + ":00")19        .otherwise(pl.col("EXSTDTM")),20        EXDOSE=pl.col("EXDOSE").cast(pl.Float64, strict=False),21    )22    # Chronological treatment-start order within each subject, breaking23    # timestamp ties by sequence number. ISO datetimes sort as text.24    .sort(["USUBJID", "EXSTDTM", "EXSEQ"])25    .with_columns(PREVDOSE=pl.col("EXDOSE").shift(1).over("USUBJID"))26    # Y when the current dose is lower than the previous one; no value for27    # the first record and whenever either dose is zero or missing. The28    # previous dose is always the administration just before.29    .with_columns(30        DOSREDFL=pl.when(31            pl.col("EXDOSE").is_not_null()32            & (pl.col("EXDOSE") != 0)33            & pl.col("PREVDOSE").is_not_null()34            & (pl.col("PREVDOSE") != 0)35            & (pl.col("EXDOSE") < pl.col("PREVDOSE"))36        )37        .then(pl.lit("Y"))38        .otherwise(None)39    )40    .select("STUDYID", "USUBJID", "EXSEQ", "EXSTDTM", "EXDOSE", "DOSREDFL")41)4243Path("/app/output").mkdir(exist_ok=True)44adex.write_csv("/app/output/adex.csv")

1/* opensas reference solution for the yamaa benchmark adam-adex-dose-reduction.2   Reads the benchmark inputs and derives the requested output.3   Execute with the existing opensas runtime: opensas /app/output/result.sas. */45data ex;6  length STUDYID USUBJID EXSTDTM $1024;7  infile "/app/input/ex.csv" dsd dlm="," firstobs=2 truncover lrecl=32767;8  input STUDYID : $char1024. USUBJID : $char1024. EXSEQ EXSTDTM : $char1024. EXDOSE;9run;1011proc sort data=ex out=ordered; by STUDYID USUBJID EXSTDTM EXSEQ; run;12data result;13  length DOSREDFL $1;14  retain previous;15  set ordered; by STUDYID USUBJID;16  if first.USUBJID then previous=.;17  if EXDOSE>0 and previous>0 and EXDOSE<previous then DOSREDFL='Y';18  previous=EXDOSE;19run;2021proc sql;22  create table final as select STUDYID, USUBJID, EXSEQ, EXSTDTM, EXDOSE, DOSREDFL from result;23quit;24proc export data=final outfile="/app/output/adex.csv" dbms=csv replace; run;