From d7ce47c13f4844c6fe5f89e25fe9a005cb525f86 Mon Sep 17 00:00:00 2001
From: ldam <ldam@passcal.nmt.edu>
Date: Tue, 26 Sep 2023 16:24:42 -0600
Subject: [PATCH] flake8, type hint

---
 .../view/plotting/plotting_widget/plotting.py | 41 ++++++++++---------
 1 file changed, 21 insertions(+), 20 deletions(-)

diff --git a/sohstationviewer/view/plotting/plotting_widget/plotting.py b/sohstationviewer/view/plotting/plotting_widget/plotting.py
index 129772d5a..f0249f0f5 100644
--- a/sohstationviewer/view/plotting/plotting_widget/plotting.py
+++ b/sohstationviewer/view/plotting/plotting_widget/plotting.py
@@ -1,5 +1,5 @@
 # class with all plotting functions
-from typing import Dict, Optional
+from typing import Dict
 import numpy as np
 from matplotlib.axes import Axes
 
@@ -100,6 +100,24 @@ class Plotting:
         ax.chan_db_info = chan_db_info
         return ax
 
+    def plot_multi_color_dots_equal_on_upper_bound(
+            self, c_data: Dict, chan_db_info: Dict, chan_id: str) -> Axes:
+        """
+        Use plot_multi_color_dots_base() to plot channel in which colors are
+        identified by plot_from_value_color_equal_on_upper_bound
+        """
+        return self.plot_multi_color_dots_base(
+            c_data, chan_db_info, equal_upper=True)
+
+    def plot_multi_color_dots_equal_on_lower_bound(
+            self, c_data: Dict, chan_db_info: Dict, chan_id: str) -> Axes:
+        """
+        Use plot_multi_color_dots_base() to plot channel in which colors are
+        identified by plot_from_value_color_equal_on_lower_bound
+        """
+        return self.plot_multi_color_dots_base(
+            c_data, chan_db_info, equal_upper=False)
+
     def plot_tri_colors(
             self, c_data: Dict, chan_db_info: Dict, chan_id: str) -> Axes:
         """
@@ -163,25 +181,8 @@ class Plotting:
         ax.chan_db_info = chan_db_info
         return ax
 
-    def plot_multi_color_dots_equal_on_upper_bound(
-            self, c_data, chan_db_info, chan_id):
-        """
-        Use plot_multi_color_dots_base() to plot channel in which colors are
-        identified by plot_from_value_color_equal_on_upper_bound
-        """
-        return self.plot_multi_color_dots_base(
-            c_data, chan_db_info, equal_upper=True)
-
-    def plot_multi_color_dots_equal_on_lower_bound(
-            self, c_data, chan_db_info, chan_id):
-        """
-        Use plot_multi_color_dots_base() to plot channel in which colors are
-        identified by plot_from_value_color_equal_on_lower_bound
-        """
-        return self.plot_multi_color_dots_base(
-            c_data, chan_db_info, equal_upper=False)
-
-    def plot_up_down_dots(self, c_data, chan_db_info, chan_id):
+    def plot_up_down_dots(
+            self, c_data: Dict, chan_db_info: Dict, chan_id: str) -> Axes:
         """
         Plot channel with 2 different values, one above, one under center line.
         Each value has corresponding color defined in valueColors in database.
-- 
GitLab