From a3941e76c43554df8049215ecf439392bc11df1c Mon Sep 17 00:00:00 2001
From: lynnschmittwilken <l.schmittwilken@tu-berlin.de>
Date: Mon, 15 May 2023 14:25:14 -0400
Subject: [PATCH] fix: aligned functions args and keys in output dicts

---
 stimupy/stimuli/checkerboards.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/stimupy/stimuli/checkerboards.py b/stimupy/stimuli/checkerboards.py
index 3be3ad0..9a966ba 100644
--- a/stimupy/stimuli/checkerboards.py
+++ b/stimupy/stimuli/checkerboards.py
@@ -292,6 +292,9 @@ def checkerboard(
         "period": period,
         "rotation": rotation,
         "intensity_checks": intensity_checks,
+        "target_indices": target_indices,
+        "extend_targets": extend_targets,
+        "round_phase_width": round_phase_width,
     }
 
     # Add targets
@@ -414,6 +417,10 @@ def contrast_contrast(
 
     stim["img"] = img
     stim["target_mask"] = mask.astype(int)
+    stim["target_shape"] = target_shape
+    stim["alpha"] = alpha
+    stim["tau"] = tau
+    stim["round_phase_width"] = round_phase_width
     return stim