Merge pull request #1007 from NREL-Sienna/jd/handle_1d_DAxArr

add missing method
NREL-Sienna · Sep 13, 2023 · e69e29d · e69e29d · jd-lara · Sep 13, 2023
2 parents 0f4628b + 0ed7998
commit e69e29d
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/operation/decision_model_store.jl b/src/operation/decision_model_store.jl
@@ -77,6 +77,25 @@ function write_result!(
     return
 end
 
+function write_result!(
+    store::DecisionModelStore,
+    name::Symbol,
+    key::OptimizationContainerKey,
+    index::DecisionModelIndexType,
+    update_timestamp::Dates.DateTime,
+    array::DenseAxisArray{<:Any, 1},
+)
+    columns = axes(array)[1]
+    if eltype(columns) !== String
+        # TODO: This happens because buses are stored by indexes instead of name.
+        columns = string.(columns)
+    end
+    container = getfield(store, get_store_container_type(key))
+    container[key][index] =
+        DenseAxisArray(reshape(array.data, 1, length(columns)), ["1"], columns)
+    return
+end
+
 function read_results(
     store::DecisionModelStore,
     key::OptimizationContainerKey;