diff --git a/sample_data/meds_sample/held_out/0.parquet b/sample_data/meds_sample/held_out/0.parquet
index e7a31e5..0dead93 100644
Binary files a/sample_data/meds_sample/held_out/0.parquet and b/sample_data/meds_sample/held_out/0.parquet differ
diff --git a/sample_data/meds_sample/sample_shard.parquet b/sample_data/meds_sample/sample_shard.parquet
index b69f923..e31ab77 100644
Binary files a/sample_data/meds_sample/sample_shard.parquet and b/sample_data/meds_sample/sample_shard.parquet differ
diff --git a/sample_data/meds_sample/train/0.parquet b/sample_data/meds_sample/train/0.parquet
index d3f6f2b..414a315 100644
Binary files a/sample_data/meds_sample/train/0.parquet and b/sample_data/meds_sample/train/0.parquet differ
diff --git a/sample_data/meds_sample/train/1.parquet b/sample_data/meds_sample/train/1.parquet
index dd79221..bbd7a3e 100644
Binary files a/sample_data/meds_sample/train/1.parquet and b/sample_data/meds_sample/train/1.parquet differ
diff --git a/sample_data/meds_sample/tuning/0.parquet b/sample_data/meds_sample/tuning/0.parquet
index ad3f7ca..0b757be 100644
Binary files a/sample_data/meds_sample/tuning/0.parquet and b/sample_data/meds_sample/tuning/0.parquet differ