fixes to run

NREL-Sienna · Sep 29, 2023 · ba4e377 · ba4e377
1 parent ecf9a0a
commit ba4e377
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/simulation/hdf_simulation_store.jl b/src/simulation/hdf_simulation_store.jl
@@ -595,7 +595,7 @@ function write_result!(
     output_cache = get_output_cache(store.cache, model_name, key)
     cur_size = get_size(store.cache)
 
-    add_result!(output_cache, index, data.data, is_full(store.cache, cur_size))
+    add_result!(output_cache, index, permutedims(data.data, (3, 1, 2)), is_full(store.cache, cur_size))
 
     if get_dirty_size(output_cache) >= get_min_flush_size(store.cache)
         discard = !should_keep_in_cache(output_cache)

diff --git a/src/simulation/optimization_output_cache.jl b/src/simulation/optimization_output_cache.jl
@@ -52,7 +52,10 @@ end
 """
 Add result to the cache.
 """
-function add_result!(cache::OptimizationOutputCache, timestamp, array, system_cache_is_full)
+function add_result!(cache::OptimizationOutputCache,
+    timestamp::Dates.DateTime,
+    array::Array{Float64},
+    system_cache_is_full::Bool)
     if cache.size_per_entry == 0
         cache.size_per_entry = length(array) * sizeof(first(array))
     end
@@ -77,7 +80,11 @@ function add_result!(cache::OptimizationOutputCache, timestamp, array, system_ca
     return cache.size_per_entry
 end
 
-function _add_result!(cache::OptimizationOutputCache, timestamp, data)
+function _add_result!(
+    cache::OptimizationOutputCache,
+    timestamp::Dates.DateTime,
+    data::Array{Float64},
+)
     cache.data[timestamp] = data
     push!(cache.dirty_timestamps, timestamp)
     return