From 52b57de00d67b959508eedfb162314cdce7902dc Mon Sep 17 00:00:00 2001
From: Joris Vincent <joris.vincent@gmail.com>
Date: Wed, 22 Mar 2023 14:23:23 +0100
Subject: [PATCH] Bugfix: output visual_size, ppd in `gratings.on_grating`

Closes #18
---
 stimupy/stimuli/gratings.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/stimupy/stimuli/gratings.py b/stimupy/stimuli/gratings.py
index 3c0f8ff5..9be695c4 100644
--- a/stimupy/stimuli/gratings.py
+++ b/stimupy/stimuli/gratings.py
@@ -198,7 +198,11 @@ def on_grating_masked(
 
     stim = {
         "img": img,
+        "visual_size": large_grating["visual_size"],
+        "ppd": large_grating["ppd"],
         "target_mask": mask.astype(int),
+        "small_grating_mask": small_grating["grating_mask"],
+        "large_grating_mask": large_grating["grating_mask"],
         "bar_width_small": small_grating["bar_width"],
         "bar_width_large": large_grating["bar_width"],
     }