Fix to output type for partitioned arrays

sintefmath · Mar 4, 2024 · 004e59b · 004e59b
1 parent f11ef65
commit 004e59b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ext/JutulPartitionedArraysExt/io.jl b/ext/JutulPartitionedArraysExt/io.jl
@@ -43,13 +43,14 @@ function rank_folder(pth, rank)
 end
 
 function consolidate_distributed_results(states, reports, partitions)
-    state = JUTUL_OUTPUT_TYPE()
-    report = JUTUL_OUTPUT_TYPE()
+    JT = Jutul.JUTUL_OUTPUT_TYPE
+    state = JT()
+    report = JT()
     mlabel = partitions[1]["main_partition_label"]
     if isnothing(mlabel)
         consolidate_cell_values!(state, states, partitions)
     else
-        main_state = JUTUL_OUTPUT_TYPE()
+        main_state = JT()
         for substate in states
             for (k, v) in pairs(substate)
                 if k == mlabel