Minor cleanup

computational-psychology · Dec 14, 2022 · 7f7a188 · 7f7a188
1 parent 83d37a3
commit 7f7a188
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/stimuli/components/angular.py b/stimuli/components/angular.py
@@ -40,8 +40,7 @@ def mask_angle(
         and additional params
     """
 
-    # Create mask
-    stim = mask_elements(
+    return mask_elements(
         edges=np.deg2rad(angles),
         orientation="angular",
         rotation=rotation,
@@ -50,8 +49,6 @@ def mask_angle(
         ppd=ppd,
     )
 
-    return stim
-
 
 def wedge(
     width,

diff --git a/stimuli/components/frame.py b/stimuli/components/frame.py
@@ -34,6 +34,7 @@ def mask_frames(
         mask with integer index for each frame (key: "mask"),
         and additional keys containing stimulus parameters
     """
+
     return mask_elements(
         orientation="cityblock",
         edges=edges,