fixed error in which the treatment levels for the dicitonary were not…

… sorted
TARGENE · Jul 30, 2024 · aa7c1fb · aa7c1fb
1 parent 3e44f07
commit aa7c1fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/counterfactual_mean_based/estimands.jl b/src/counterfactual_mean_based/estimands.jl
@@ -252,7 +252,7 @@ Dictionary to avoid compilation bottlenecks.
 function unique_treatment_values_dict(dataset, treatments)
     sorted_dict = Dict{Symbol, Vector{UInt8}}()
     for colname in keys(treatments)
-        sorted_dict[colname] = get_treatment_values(dataset, colname)
+        sorted_dict[colname] = sort(get_treatment_values(dataset, colname))
     end
     return sorted_dict
 end