FEAT: Incorporate FIndex in index matrix demo.

casact · Jan 10, 2025 · a4d59cf · a4d59cf
1 parent 2bbcaed
commit a4d59cf
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/faslr/demos/index/index_matrix_demo.py b/faslr/demos/index/index_matrix_demo.py
@@ -7,7 +7,7 @@
 import sys
 
 from faslr.index import (
-    index_matrix,
+    FIndex,
     IndexMatrixWidget
 )
 
@@ -25,19 +25,15 @@
 if TYPE_CHECKING:
     from pandas import DataFrame
 
-rate_factors = [
-    x + 1 for x in XYZ_RATE_CHANGES
-]
-
-df_rl: DataFrame = index_matrix(
-    years=XYZ_SAMPLE_YEARS,
-    index=rate_factors
+index = FIndex(
+    origin=XYZ_SAMPLE_YEARS,
+    changes=XYZ_RATE_CHANGES
 )
 
 app = QApplication(sys.argv)
 
 index_matrix_widget = IndexMatrixWidget(
-    matrix=df_rl
+    matrix=index.matrix
 )
 index_matrix_widget.setWindowTitle("Index Matrix Demo")