Activate model env before calling RelationshipClass and Parameter

src/data_structure/stochastic_structure.jl

@@ -511,17 +511,25 @@ function node_investment_stochastic_time_indices(
 end
 
 function node_stochastic_scenario_weight(m; kwargs...)
-    m.ext[:spineopt].stochastic_structure[:node_stochastic_scenario_weight][(; kwargs...)]
+    _with_model_env(m) do
+        m.ext[:spineopt].stochastic_structure[:node_stochastic_scenario_weight][(; kwargs...)]
+    end
 end
 
 function unit_stochastic_scenario_weight(m; kwargs...)
-    m.ext[:spineopt].stochastic_structure[:unit_stochastic_scenario_weight][(; kwargs...)]
+    _with_model_env(m) do
+        m.ext[:spineopt].stochastic_structure[:unit_stochastic_scenario_weight][(; kwargs...)]
+    end
 end
 
 function connection_stochastic_scenario_weight(m; kwargs...)
-    m.ext[:spineopt].stochastic_structure[:connection_stochastic_scenario_weight][(; kwargs...)]
+    _with_model_env(m) do
+        m.ext[:spineopt].stochastic_structure[:connection_stochastic_scenario_weight][(; kwargs...)]
+    end
 end
 
 function any_stochastic_scenario_weight(m; kwargs...)
-    m.ext[:spineopt].stochastic_structure[:any_stochastic_scenario_weight][(; kwargs...)]
+    _with_model_env(m) do
+        m.ext[:spineopt].stochastic_structure[:any_stochastic_scenario_weight][(; kwargs...)]
+    end
 end

src/data_structure/temporal_structure.jl

@@ -555,7 +555,11 @@ the second starting when the first ends.
   - `t_before`: if given, return an `Array` of `TimeSlice`s that start when `t_before` ends.
   - `t_after`: if given, return an `Array` of `TimeSlice`s that end when `t_after` starts.
 """
-t_before_t(m::Model; kwargs...) = m.ext[:spineopt].temporal_structure[:t_before_t](; kwargs...)
+function t_before_t(m::Model; kwargs...)
+    _with_model_env(m) do
+        m.ext[:spineopt].temporal_structure[:t_before_t](; kwargs...)
+    end
+end
 
 """
     t_in_t(m; t_short=anything, t_long=anything)
@@ -567,7 +571,11 @@ the second containing the first.
   - `t_short`: if given, return an `Array` of `TimeSlice`s that contain `t_short`.
   - `t_long`: if given, return an `Array` of `TimeSlice`s that are contained in `t_long`.
 """
-t_in_t(m::Model; kwargs...) = m.ext[:spineopt].temporal_structure[:t_in_t](; kwargs...)
+function t_in_t(m::Model; kwargs...)
+    _with_model_env(m) do
+        m.ext[:spineopt].temporal_structure[:t_in_t](; kwargs...)
+    end
+end
 
 """
     t_in_t_excl(m; t_short=anything, t_long=anything)
@@ -578,7 +586,11 @@ Same as [t_in_t](@ref) but exclude tuples of the same `TimeSlice`.
   - `t_short`: if given, return an `Array` of `TimeSlice`s that contain `t_short` (other than `t_short` itself).
   - `t_long`: if given, return an `Array` of `TimeSlice`s that are contained in `t_long` (other than `t_long` itself).
 """
-t_in_t_excl(m::Model; kwargs...) = m.ext[:spineopt].temporal_structure[:t_in_t_excl](; kwargs...)
+function t_in_t_excl(m::Model; kwargs...)
+    _with_model_env(m) do
+        m.ext[:spineopt].temporal_structure[:t_in_t_excl](; kwargs...)
+    end
+end
 
 """
     t_overlaps_t(m; t)

src/util/misc.jl

@@ -359,6 +359,14 @@ _percentage_str(x::Number) = string(@sprintf("%1.4f", x * 100), "%")
 
 _number_str(x::Number) = @sprintf("%.5e", x)
 
+function _with_model_env(f, m)
+    st = m.ext[:spineopt].stage
+    st === nothing && return f()
+    with_env(stage_scenario(stage=st)) do
+        f()
+    end
+end
+
 # Base
 _ObjectArrayLike = Union{ObjectLike,Array{T,1} where T<:ObjectLike}
 _RelationshipArrayLike{K} = NamedTuple{K,V} where {K,V<:Tuple{Vararg{_ObjectArrayLike}}}