fixing default arguments

b-vanstraaten · Jul 18, 2024 · c0bded9 · c0bded9
1 parent 368657e
commit c0bded9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/examples/gui.py b/examples/gui.py
@@ -12,8 +12,8 @@
 
 Cdd = [
     [0., 0.2, 0.2, 0.5],
-    [0.2, 0., 0.5, 0.2],
-    [0.2, 0.5, 0.2, 0.],
+    [0.2, 0., 0.05, 0.2],
+    [0.2, 0.05, 0.2, 0.],
     [0.5, 0.2, 0., 0]
 ]
 
@@ -34,4 +34,4 @@
     implementation='rust', charge_carrier='h', T=0., threshold=1.,
     max_charge_carriers=4,
 )
-model.run_gui(plot = 'colour_map', cmap = 'viridis')
+model.run_gui(plot = 'changes')
diff --git a/qarray/DotArrays/DotArray.py b/qarray/DotArrays/DotArray.py
@@ -188,7 +188,7 @@ def compute_threshold_estimate(self):
         """
         return compute_threshold(self.cdd)
 
-    def run_gui(self, port=27182, print_compute_time: bool = False, plot = 'changes', cmap = 'viridis'):
+    def run_gui(self, port=27182, print_compute_time: bool = False, plot = 'changes', cmap = None):
         """
         Creates a GUI for the dot array
         """

diff --git a/qarray/gui/gui.py b/qarray/gui/gui.py
@@ -286,9 +286,8 @@ def update(Cdd, Cgd, x_gate, x_amplitude, x_resolution, y_gate, y_amplitude, y_r
         match plot:
             case 'changes':
                 z = dot_occupation_changes(n).astype(float)
-
                 if cmap is None:
-                    cmap = 'gray'
+                    cmap = 'greys'
 
             case 'colour_map':
                 z = charge_state_to_unique_index(n).astype(float)