Changes when figure_kwargs is defined for cleaner conditional logic

apasarkar · apasarkar · commit 90af0d4a136d · 2024-07-16T12:12:08.000-04:00
diff --git a/fastplotlib/widgets/image.py b/fastplotlib/widgets/image.py
@@ -357,6 +357,9 @@ def __init__(
         """
         self._names = None
 
+        if figure_kwargs is None:
+            figure_kwargs = dict()
+
         # output context
         self._output = None
 
@@ -368,9 +371,8 @@ def __init__(
             if all([_is_arraylike(d) for d in data]):
                 # Grid computations
                 if figure_shape is None:
-                    if figure_kwargs is not None:
-                        if "shape" in figure_kwargs:
-                            figure_shape = figure_kwargs["shape"]
+                    if "shape" in figure_kwargs:
+                        figure_shape = figure_kwargs["shape"]
                     else:
                         figure_shape = calculate_figure_shape(len(data))
 
@@ -508,8 +510,6 @@ def __init__(
                         )
 
         figure_kwargs_default = {"controller_ids": "sync"}
-        if figure_kwargs is None:
-            figure_kwargs = dict()
 
         # update the default kwargs with any user-specified kwargs
         # user specified kwargs will overwrite the defaults
