Updating logging_times to include start_time and end_time in ensemble…

…_sample
ciemss · Apr 15, 2024 · 4798686 · 4798686
1 parent b000bec
commit 4798686
Show file tree

Hide file tree

Showing 2 changed files with 251 additions and 250 deletions.
diff --git a/docs/source/interfaces.ipynb b/docs/source/interfaces.ipynb
diff --git a/pyciemss/interfaces.py b/pyciemss/interfaces.py
@@ -146,9 +146,10 @@ def ensemble_sample(
             model_paths_or_jsons, dirichlet_alpha, solution_mappings
         )
 
-        logging_times = torch.arange(
-            start_time + logging_step_size, end_time, logging_step_size
-        )
+        # logging_times = torch.arange(
+        #     start_time + logging_step_size, end_time, logging_step_size
+        # )
+        logging_times = torch.arange(start_time, end_time + logging_step_size, logging_step_size)
 
         # Check that num_samples is a positive integer
         if not (isinstance(num_samples, int) and num_samples > 0):