FEAT: Begin adding annotations to average.py.

Author: genedan

faslr/common/model.py

@@ -473,8 +473,7 @@ def __init__(
         # Dialog box that pops up when you click the add average button.
         self.average_box = FAverageBox(
             parent=self,
-            data=averages,
-            selection_model=self.parent.selection_model
+            data=averages
         )
 
         self.layout.addWidget(self.add_average_button)

faslr/methods/expected_loss.py

@@ -210,7 +210,7 @@ def __init__(
         self.apriori_tab = QWidget()
 
 
-        self.main_tabs.addTab(self.indexation, "Indexation")
+        self.main_tabs.addTab(self.indexation, "Indexes")
         self.main_tabs.addTab(self.apriori_tab, "Apriori Selection")
 
         self.apriori_view = ExpectedLossView()

faslr/model/average.py

@@ -37,7 +37,7 @@
 if TYPE_CHECKING:
     from pandas import DataFrame
     from faslr.common.model import (
-        FSelectionModel,
+        FSelectionModelToolbox,
         FSelectionModelWidget
     )
 
@@ -49,18 +49,15 @@ class FAverageBox(QDialog):
     Parameters
     ----------
     parent: Optional[FSelectionModelWidget]
-        The containing widget of the corresponding selection model.
-    selection_model: Optional[FSelectionModel]
-        The corresponding selection model.
+        The containing toolbox of the corresponding selection model.
     title: Optional[str]
         The title of the dialog box, overriding the default.
     data: Optional[DataFrame]
         A dataframe containing the available averages. Overrides extraction of these averages from the database.
     """
     def __init__(
             self,
-            parent: Optional[FSelectionModelWidget],
-            selection_model: Optional[FSelectionModel] = None,
+            parent: Optional[FSelectionModelToolbox],
             title: Optional[str] = "Available Averages",
             data: Optional[DataFrame] = None
     ):
@@ -69,9 +66,9 @@ def __init__(
 
         self.layout = QVBoxLayout()
 
-        self.parent: FSelectionModelWidget = parent
+        self.parent: FSelectionModelWidget = parent.parent
 
-        self.selection_model = selection_model
+        self.selection_model = self.parent.selection_model
 
         self.model = FAverageModel(
             parent=None,