fixed bug that parallelogram and ellipse would not contain correct si…

…ze params in output dict
computational-psychology · Feb 22, 2023 · 92acc3c · 92acc3c
1 parent 33e0c02
commit 92acc3c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/stimupy/components/shapes.py b/stimupy/components/shapes.py
@@ -431,9 +431,9 @@ def parallelogram(
     return {
         "img": img * (intensity_parallelogram - intensity_background) + intensity_background,
         "shape_mask": img.astype(int),
-        "shape": shape,
-        "visual_size": visual_size,
-        "ppd": ppd,
+        "shape": base["shape"],
+        "visual_size": base["visual_size"],
+        "ppd": base["ppd"],
         "parallelogram_size": parallelogram_size,
         "intensity_background": intensity_background,
         "intensity_parallelogram": intensity_parallelogram,
@@ -520,9 +520,9 @@ def ellipse(
     return {
         "img": img * (intensity_ellipse - intensity_background) + intensity_background,
         "shape_mask": img.astype(int),
-        "shape": shape,
-        "visual_size": visual_size,
-        "ppd": ppd,
+        "shape": base["shape"],
+        "visual_size": base["visual_size"],
+        "ppd": base["ppd"],
         "radius": radius,
         "intensity_background": intensity_background,
         "intensity_ellipse": intensity_ellipse,