Always have context in MultiModel

sintefmath · Nov 20, 2024 · 2094596 · 2094596
1 parent 318c1ac
commit 2094596
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/core_types/core_types.jl b/src/core_types/core_types.jl
@@ -965,7 +965,17 @@ struct MultiModel{label, T, CT, G, C, GL} <: AbstractMultiModel{label}
     group_lookup::GL
 end
 
-function MultiModel(models, label::Union{Nothing, Symbol} = nothing; cross_terms = Vector{CrossTermPair}(), groups = nothing, context = nothing, reduction = missing, specialize = false, specialize_ad = false)
+function MultiModel(models, label::Union{Nothing, Symbol} = nothing;
+        cross_terms = Vector{CrossTermPair}(),
+        groups = nothing,
+        context = nothing,
+        reduction = missing,
+        specialize = false,
+        specialize_ad = false
+    )
+    if isnothing(context)
+        context = models[first(keys(models))].context
+    end
     group_lookup = Dict{Symbol, Int}()
     if isnothing(groups)
         num_groups = 1