Add orientation to square_wave component

computational-psychology · Dec 6, 2022 · 9ff5628 · 9ff5628
1 parent 3d75063
commit 9ff5628
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/stimuli/components/grating.py b/stimuli/components/grating.py
@@ -249,6 +249,7 @@ def square_wave(
     frequency=None,
     n_bars=None,
     bar_width=None,
+    orientation="horizontal",
     intensity_bars=(0.0, 1.0),
     period="ignore",
 ):
@@ -287,6 +288,7 @@ def square_wave(
         frequency=frequency,
         n_bars=n_bars,
         bar_width=bar_width,
+        orientation=orientation,
         period=period,
     )
     mask = stim["mask"]