Update widget.py

nglviewer · Sep 5, 2023 · d3e0525 · d3e0525
1 parent 1a92189
commit d3e0525
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/nglview/widget.py b/nglview/widget.py
@@ -189,12 +189,15 @@ def __init__(self,
                  structure=None,
                  representations=None,
                  parameters=None,
+                 gui=False,
+                 theme='default',
+                 default=False,
                  **kwargs):
         super().__init__(**kwargs)
 
         self._gui = None
-        self._init_gui = kwargs.pop('gui', False)
-        self._theme = kwargs.pop('theme', 'default')
+        self._init_gui = gui
+        self._theme = theme
         self._widget_image = Image()
         self._widget_image.width = 900.
         self._image_array = []
@@ -224,8 +227,8 @@ def __init__(self,
             # initial representations will be set.
             kwargs['default_representation'] = False
         else:
-            if 'default' in kwargs:
-                kwargs['default_representation'] = kwargs['default']
+            if default:
+                kwargs['default_representation'] = default
 
         autoview = 'center' not in kwargs or ('center' in kwargs
                                               and kwargs.pop('center'))