From 67d723ab5961845394d129666b72f5b0925b0be8 Mon Sep 17 00:00:00 2001 From: Florian Wiesner Date: Mon, 19 May 2025 16:40:41 -0400 Subject: [PATCH 1/2] added max rollout steps to dataset docstring --- the_well/data/datasets.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/the_well/data/datasets.py b/the_well/data/datasets.py index c3b82593..e2b5f8ba 100644 --- a/the_well/data/datasets.py +++ b/the_well/data/datasets.py @@ -136,6 +136,9 @@ class WellDataset(Dataset): Whether to normalize data in the dataset normlization_type: What type of dataset normalization. Callable Options: ZSCORE and RMS + max_rollout_steps: + Maximum number of output steps to return in a single sample, mostly used for + full trajectory mode. n_steps_input: Number of steps to include in each sample n_steps_output: From 49982f12deaa2eeed604913f0d38ae5cdc938f72 Mon Sep 17 00:00:00 2001 From: Florian Wiesner <65916600+FloWsnr@users.noreply.github.com> Date: Wed, 21 May 2025 10:11:53 -0400 Subject: [PATCH 2/2] Update the_well/data/datasets.py Co-authored-by: Lucas Meyer --- the_well/data/datasets.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/the_well/data/datasets.py b/the_well/data/datasets.py index e2b5f8ba..7620c788 100644 --- a/the_well/data/datasets.py +++ b/the_well/data/datasets.py @@ -137,8 +137,7 @@ class WellDataset(Dataset): normlization_type: What type of dataset normalization. Callable Options: ZSCORE and RMS max_rollout_steps: - Maximum number of output steps to return in a single sample, mostly used for - full trajectory mode. + Maximum number of output steps to return in a single sample. Return the full trajectory if larger than its actual length. n_steps_input: Number of steps to include in each sample n_steps_output: