after merge

Author: pat-alt

Project.toml

@@ -1,7 +1,7 @@
 name = "AlgorithmicRecourseDynamics"
 uuid = "3d1ede72-abb8-4340-bf8e-2ae06849b5ec"
 authors = ["Patrick Altmeyer"]
-version = "0.1.0"
+version = "0.1.1"
 
 [deps]
 CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b"
@@ -27,6 +27,7 @@ StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 
 [compat]
 CSV = "0.10"
+CounterfactualExplanations = "0.1"
 DataFrames = "1"
 Distances = "0.10"
 Flux = "0.13"

src/experiments/functions.jl

@@ -173,12 +173,13 @@ function update_experiment!(experiment::Experiment, recourse_system::RecourseSys
         X′ = reduce(hcat, @.(selectdim(counterfactual(results), 3, indices_)))
         y′ = reduce(hcat, @.(selectdim(counterfactual_label(results), 3, indices_)))
 
-        # If for any counterfactuals the returned label is NaN, this is considered as invalid and the current label is not updated:
-        chosen_individuals = chosen_individuals[vec(.!(isnan.(y′)))]
+    # If for any counterfactuals the returned label is NaN, this is considered as invalid and the current label is not updated:
+    valid_ces = vec(.!(isnan.(y′)))
+    chosen_individuals = chosen_individuals[valid_ces]
 
-        # Update data:
-        X[:, chosen_individuals] = X′
-        y[:, chosen_individuals] = y′
+    # Update data:
+    X[:, chosen_individuals] = X′[:, valid_ces]
+    y[:, chosen_individuals] = y′[:, valid_ces]
 
         # Generative model:
         gen_mod = deepcopy(counterfactual_data.generative_model)