🐛 fix minor bug (wrongly assigned feat)

RasmussenLab · Jul 9, 2024 · 58f08e4 · 58f08e4
1 parent 1c72316
commit 58f08e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/move/data/perturbations.py b/src/move/data/perturbations.py
@@ -313,8 +313,8 @@ def perturb_continuous_data_extended(
         perturbations_list.append(perturbed_con[:, start_idx + i].numpy())
 
         perturbed_dataloader = _build_dataloader(
-            con_data=baseline_dataset.cat_all,
-            cat_data=perturbed_con,
+            cat_data=baseline_dataset.cat_all,
+            con_data=perturbed_con,
             cat_shapes=baseline_dataset.cat_shapes,
             con_shapes=baseline_dataset.con_shapes,
             batch_size=baseline_dataloader.batch_size,