FEAT: Continue work on incorporating index matrix into expected loss …

…model.
casact · Jan 8, 2025 · 4bf9cda · 4bf9cda
1 parent daca3b5
commit 4bf9cda
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/faslr/index/index_matrix.py b/faslr/index/index_matrix.py
@@ -113,7 +113,7 @@ def __init__(self):
 class IndexMatrixWidget(QWidget):
     def __init__(
             self,
-            matrix: DataFrame
+            matrix: DataFrame = None
     ):
         super().__init__()
 

diff --git a/faslr/methods/expected_loss.py b/faslr/methods/expected_loss.py
@@ -23,6 +23,8 @@
 from faslr.index import (
     calculate_index_factors,
     IndexInventory,
+    IndexMatrixModel,
+    IndexMatrixView,
     IndexTableModel,
     IndexTableView
 )
@@ -390,9 +392,11 @@ def __init__(
         self.index_view = IndexTableView()
         self.index_view.setModel(self.index_model)
 
-        self.dummytab = QWidget()
+        self.matrix_preview = IndexMatrixView()
+        self.matrix_model = IndexMatrixModel(matrix=None)
+
         self.index_preview.addTab(self.index_view, "Column")
-        self.index_preview.addTab(self.dummytab, "Matrix")
+        self.index_preview.addTab(self.matrix_preview, "Matrix")
 
         index_view_layout.addWidget(self.index_preview)
         index_view_container.setContentsMargins(